aek-auto-mlbuilder 0.1.1__tar.gz → 0.3.0__tar.gz

{aek_auto_mlbuilder-0.1.1 → aek_auto_mlbuilder-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.1.1
+Version: 0.3.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.3.0/aek_auto_mlbuilder/__init__.py

@@ -0,0 +1,5 @@
+from .base import BaseModel
+from .utils import split_data
+from .linear_regression import LinearRegressor
+from .logistic_regression import LogisticClassifier
+from .decision_tree import DecisionTreeModel

aek_auto_mlbuilder-0.3.0/aek_auto_mlbuilder/decision_tree.py

@@ -0,0 +1,45 @@
+from sklearn.tree import DecisionTreeClassifier, DecisionTreeRegressor
+from .base import BaseModel
+
+
+class DecisionTreeModel(BaseModel):
+    """
+    Decision Tree for classification or regression
+    Use "task" param to specify "classification" or "regression"
+    Brute force search included
+    """
+
+    def __init__(self, task="classification", param_grid=None):
+        super().__init__()
+        self.task = task
+        self.param_grid = param_grid or {
+            "max_depth": [None, 3, 5, 10],
+            "min_samples_split": [2, 5, 10],
+            "min_samples_leaf": [1, 2, 4]
+        }
+
+    def train(self, X, y):
+        best_score = -float("inf")
+        best_model = None
+
+        if self.task.lower() == "classification":
+            ModelClass = DecisionTreeClassifier
+        else:
+            ModelClass = DecisionTreeRegressor
+        
+        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(
+                        max_depth=max_depth,
+                        min_samples_split=min_samples_split,
+                        min_samples_leaf=min_samples_leaf
+                    )
+                    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.3.0/aek_auto_mlbuilder/logistic_regression.py

@@ -0,0 +1,48 @@
+from sklearn.linear_model import LogisticRegression
+from sklearn.preprocessing import StandardScaler
+from sklearn.pipeline import make_pipeline
+from .base import BaseModel
+
+
+
+class LogisticClassifier(BaseModel):
+    """
+    Basic Logistic Regression class for binary/multi-class classification.
+    Brute-force parameter search included.
+    """
+
+    def __init__(self, param_grid=None):
+        super().__init__()
+        self.param_grid = param_grid or {
+            "C": [0.01, 0.1, 1, 10],
+            "penalty": ["l2"],
+            "solver": ["lbfgs"],
+            "fit_intercept": [True, False]
+        }
+    def train(self, X, y):
+        best_score = -float("inf")
+        best_model = None
+
+        for C in self.param_grid["C"]:
+            for penalty in self.param_grid["penalty"]:
+                for solver in self.param_grid["solver"]:
+                    for fit_intercept in self.param_grid["fit_intercept"]:
+                        model = make_pipeline(
+                            StandardScaler(),
+                            LogisticRegression(
+                                C=C,
+                                penalty=penalty,
+                                solver=solver,
+                                fit_intercept=fit_intercept,
+                                max_iter=1000
+                            )
+                        )
+                        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.1.1 → aek_auto_mlbuilder-0.3.0}/aek_auto_mlbuilder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.1.1
+Version: 0.3.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.1.1 → aek_auto_mlbuilder-0.3.0}/aek_auto_mlbuilder.egg-info/SOURCES.txt

@@ -4,7 +4,9 @@ pyproject.toml
 setup.py
 aek_auto_mlbuilder/__init__.py
 aek_auto_mlbuilder/base.py
+aek_auto_mlbuilder/decision_tree.py
 aek_auto_mlbuilder/linear_regression.py
+aek_auto_mlbuilder/logistic_regression.py
 aek_auto_mlbuilder/utils.py
 aek_auto_mlbuilder.egg-info/PKG-INFO
 aek_auto_mlbuilder.egg-info/SOURCES.txt

{aek_auto_mlbuilder-0.1.1 → aek_auto_mlbuilder-0.3.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aek-auto-mlbuilder",        # PyPI dağıtım adı
-    version="0.1.1",
+    version="0.3.0",
     description="Automatic ML model builder in Python",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

aek_auto_mlbuilder-0.1.1/aek_auto_mlbuilder/__init__.py

@@ -1,3 +0,0 @@
-from .base import BaseModel
-from .utils import split_data
-from .linear_regression import LinearRegressor