aek-auto-mlbuilder 0.4.0__py3-none-any.whl → 0.6.0__py3-none-any.whl

aek_auto_mlbuilder/__init__.py

@@ -3,4 +3,6 @@ from .utils import split_data
 from .linear_regression import LinearRegressor
 from .logistic_regression import LogisticClassifier
 from .decision_tree import DecisionTreeModel
-from. knn import KNNModel
+from .knn import KNNModel
+from .random_forest import RandomForestModel
+from .svm import SVMModel

aek_auto_mlbuilder/random_forest.py

@@ -0,0 +1,50 @@
+from sklearn.ensemble import RandomForestClassifier, RandomForestRegressor
+from .base import BaseModel
+
+class RandomForestModel(BaseModel):
+    """
+    Random Forest model for classification or regression
+    Use task either "classification" or "regression"
+    brute force method is using
+    """
+    def __init__(self, task="classification", param_grid=None):
+        super().__init__()
+        self.task = task
+        self.param_grid = param_grid or {
+            "n_estimators": [50, 100, 200],
+            "max_depth": [None, 5, 10],
+            "min_samples_split": [2, 5],
+            "min_samples_leaf": [1, 2]
+        }
+    def train(self, X, y):
+        best_score = -float("inf")
+        best_model = None
+
+        if self.task.lower() == "classification":
+            ModelClass = RandomForestClassifier
+        elif self.task.lower() == "regression":
+            ModelClass = RandomForestRegressor
+        else:
+            raise ValueError("task must be 'classification' or 'regression'")
+        
+        for n_estimators in self.param_grid["n_estimators"]:
+            for max_depth in self.param_grid["max_depth"]:
+                for min_samples_split in self.param_grid["min_samples_split"]:
+                    for min_samples_leaf in self.param_grid["min_samples_leaf"]:
+                        model = ModelClass(
+                            n_estimators=n_estimators,
+                            max_depth=max_depth,
+                            min_samples_split=min_samples_split,
+                            min_samples_leaf=min_samples_leaf,
+                            random_state=42,
+                            n_jobs=-1
+                        )
+                        model.fit(X, y)
+                        score = model.score(X, y)
+                        if score > best_score:
+                            best_score = score
+                            best_model = model
+        self.best_model = best_model
+        self.best_score = best_score
+        return self.best_model
+    

aek_auto_mlbuilder/svm.py

@@ -0,0 +1,47 @@
+from sklearn.svm import SVC, SVR
+from sklearn.preprocessing import StandardScaler
+from sklearn.pipeline import make_pipeline
+from .base import BaseModel
+
+
+class SVMModel(BaseModel):
+    """
+    support vector machine for classification and regression
+    use task parameter for "regression" or "classification"
+    brute force is using
+    """
+    def __init__(self, task="classification", param_grid=None):
+        super().__init__()
+        self.task = task
+        self.param_grid = param_grid or {
+            "C": [0.1, 1, 10],
+            "kernel": ["linear", "rbf"],
+            "gamma": ["scale", "auto"]
+        }
+    
+    def train(self, X, y):
+        best_score = -float("inf")
+        best_model = None
+
+        if self.task.lower() == "classification":
+            ModelClass = SVC
+        elif self.task.lower() == "regression":
+            ModelClass = SVR
+        else:
+            raise ValueError("task must be either classification or regression")
+        
+        for C in self.param_grid["C"]:
+            for kernel in self.param_grid["kernel"]:
+                for gamma in self.param_grid["gamma"]:
+                    model = make_pipeline(
+                        StandardScaler(),
+                        ModelClass(C=C, kernel=kernel, gamma=gamma)
+                    )
+                    model.fit(X, y)
+                    score = model.score(X, y)
+                    if score > best_score:
+                        best_score = score
+                        best_model = model
+        self.best_model = best_model
+        self.best_score = best_score
+        return self.best_model

{aek_auto_mlbuilder-0.4.0.dist-info → aek_auto_mlbuilder-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.4.0
+Version: 0.6.0
 Summary: Automatic ML model builder in Python
 Home-page: https://github.com/alpemre8/aek-auto-mlbuilder
 Author: Alp Emre Karaahmet

{aek_auto_mlbuilder-0.4.0.dist-info → aek_auto_mlbuilder-0.6.0.dist-info}/RECORD

@@ -1,12 +1,14 @@
-aek_auto_mlbuilder/__init__.py,sha256=qB67FHc_lJeKMQltNz087hseQUeIvd15njtNrGAqXRk,227
+aek_auto_mlbuilder/__init__.py,sha256=MNRFS2qzHSmCsl7a3l5XUaifAYF4kSuj9dquRdv3ciY,298
 aek_auto_mlbuilder/base.py,sha256=GgMdAoceRjwz3i9rVQ0RAjvn5ZdRS-sAkLWjymbE8s0,385
 aek_auto_mlbuilder/decision_tree.py,sha256=OImOHaREz2jWcANXVC6VKcatFJfzrtyCHJKXNA5-hoI,1606
 aek_auto_mlbuilder/knn.py,sha256=bNADSq2Ce6stmiWoumRgJbyCmp5SjsbvtTDq3cimKAk,1722
 aek_auto_mlbuilder/linear_regression.py,sha256=MtOSRiXDIJPd3abnz4yNT4DBtrkmvEy00Kbx4AFk4Kg,1259
 aek_auto_mlbuilder/logistic_regression.py,sha256=lp9-e9p9QrqL20DmhIJaSnra1SwyMiOdTfOMlgYsNQA,1707
+aek_auto_mlbuilder/random_forest.py,sha256=uvU4faNdoeMgzqrGJ85qchV3tRBaLFwSVKE3WPGWJ74,2018
+aek_auto_mlbuilder/svm.py,sha256=_7BfwTEHFy8iF7OxxvoGSobvosKSY4tgRlEIDlV4U5M,1614
 aek_auto_mlbuilder/utils.py,sha256=NcoM3b4Ng1Ogk3iKuz9DcMVwppGRqOLRp5g9jBCkWxY,190
-aek_auto_mlbuilder-0.4.0.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
-aek_auto_mlbuilder-0.4.0.dist-info/METADATA,sha256=dNYzCfMNcAIpg0CVk5CE2BxjoWn3TYhhb3D5NNxhhPc,1400
-aek_auto_mlbuilder-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aek_auto_mlbuilder-0.4.0.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
-aek_auto_mlbuilder-0.4.0.dist-info/RECORD,,
+aek_auto_mlbuilder-0.6.0.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
+aek_auto_mlbuilder-0.6.0.dist-info/METADATA,sha256=etjCA-t5wRXMpT1w1R9Qd2Ba8G-TLoZIu2VZ6LgChow,1400
+aek_auto_mlbuilder-0.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aek_auto_mlbuilder-0.6.0.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
+aek_auto_mlbuilder-0.6.0.dist-info/RECORD,,