aek-auto-mlbuilder 0.5.0__tar.gz → 0.6.0__tar.gz

{aek_auto_mlbuilder-0.5.0 → aek_auto_mlbuilder-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.5.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.5.0 → aek_auto_mlbuilder-0.6.0}/aek_auto_mlbuilder/__init__.py

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

aek_auto_mlbuilder-0.6.0/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.5.0 → aek_auto_mlbuilder-0.6.0}/aek_auto_mlbuilder.egg-info/PKG-INFO

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

@@ -9,6 +9,7 @@ aek_auto_mlbuilder/knn.py
 aek_auto_mlbuilder/linear_regression.py
 aek_auto_mlbuilder/logistic_regression.py
 aek_auto_mlbuilder/random_forest.py
+aek_auto_mlbuilder/svm.py
 aek_auto_mlbuilder/utils.py
 aek_auto_mlbuilder.egg-info/PKG-INFO
 aek_auto_mlbuilder.egg-info/SOURCES.txt

{aek_auto_mlbuilder-0.5.0 → aek_auto_mlbuilder-0.6.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.5.0",
+    version="0.6.0",
     description="Automatic ML model builder in Python",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",