aek-auto-mlbuilder 0.8.2__tar.gz → 0.10.0__tar.gz

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/PKG-INFO

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

@@ -7,4 +7,6 @@ from .knn import KNNModel
 from .random_forest import RandomForestModel
 from .svm import SVMModel
 from .naive_bayes import NaiveBayesModel
-from .kmeans import KMeansModel
+from .kmeans import KMeansModel
+from .dbscan import DBSCANModel
+from .agglomerative import AgglomerativeModel

aek_auto_mlbuilder-0.10.0/aek_auto_mlbuilder/agglomerative.py

@@ -0,0 +1,40 @@
+from sklearn.cluster import AgglomerativeClustering
+from .base import BaseModel
+
+class AgglomerativeModel(BaseModel):
+    """
+    Agglomerative clustering model
+    brute force search is being used
+    """
+
+    def __init(self, param_grid=None):
+        super().__init__()
+        self.param_grid = param_grid or {
+            "n_clusters": [2, 3, 4, 5, 6],
+            "linkage": ["ward", "complete", "average", "single"]
+        }
+
+    def train(self, X):
+        best_score = -float("inf")
+        best_model = None
+
+        for n_clusters in self.param_grid["n_clusters"]:
+            for linkage in self.param_grid["linkage"]:
+                model = AgglomerativeClustering(
+                    n_clusters=n_clusters,
+                    linkage=linkage
+                )
+                labels = model.fit_predict(X)
+                clusters = set(labels)
+                score = 0
+                for k in clusters:
+                    cluster_size = (labels == k).sum()
+                    score += 1 / cluster_size
+                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.10.0/aek_auto_mlbuilder/dbscan.py

@@ -0,0 +1,31 @@
+from sklearn.cluster import DBSCAN
+from .base import BaseModel
+
+class DBSCANModel(BaseModel):
+    """
+    DBSCAN clustering model (unsupervised learning)
+    brute force searching is being used
+    """
+    def __init__(self, param_grid=None):
+        super().__init__()
+        self.param_grid = param_grid or {
+            "eps": [0.3, 0.5, 0.7, 1.0],
+            "min_samples": [3, 5, 10]
+        }
+
+    def train(self, X):
+        best_score = -float("inf")
+        best_model = None
+
+        for eps in self.param_grid["eps"]:
+            for min_samples in self.param_grid["min_samples"]:
+                model = DBSCAN(eps=eps, min_samples=min_samples)
+                labels = model.fit_predict(X)
+
+                score = len(labels) - (labels == -1).sum()
+                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.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder/kmeans.py

@@ -5,7 +5,7 @@ from .base import BaseModel
 class KMeansModel(BaseModel):
     """
     K-Means clustering model (unsupervised learning)
-    Brute force method has been using
+    Brute force method is being used
     """
     def __init__(self, param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder/knn.py

@@ -8,7 +8,7 @@ class KNNModel(BaseModel):
     """
     KNN model for classification for classification or regression
     Use "task" for "classification" or "regression"
-    Brute-force search has been using
+    Brute-force search is being used
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder/naive_bayes.py

@@ -6,7 +6,7 @@ class NaiveBayesModel(BaseModel):
     """
     Naive bayes model supporting gaussianNB, multinomialNB, bernoulliNB
     use 'nb_type' param to specify the variant: gaussian, multinomial, bernoulli
-    brute force hyperparameter search has been using
+    brute force hyperparameter search is being used
     """
     def __init__(self, nb_type="gaussian", param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder/random_forest.py

@@ -5,7 +5,7 @@ class RandomForestModel(BaseModel):
     """
     Random Forest model for classification or regression
     Use task either "classification" or "regression"
-    brute force method has been using
+    brute force method is being used
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder/svm.py

@@ -8,7 +8,7 @@ class SVMModel(BaseModel):
     """
     support vector machine for classification and regression
     use task parameter for "regression" or "classification"
-    brute force has been using
+    brute force is being used
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/aek_auto_mlbuilder.egg-info/PKG-INFO

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

@@ -3,7 +3,9 @@ README.md
 pyproject.toml
 setup.py
 aek_auto_mlbuilder/__init__.py
+aek_auto_mlbuilder/agglomerative.py
 aek_auto_mlbuilder/base.py
+aek_auto_mlbuilder/dbscan.py
 aek_auto_mlbuilder/decision_tree.py
 aek_auto_mlbuilder/kmeans.py
 aek_auto_mlbuilder/knn.py

{aek_auto_mlbuilder-0.8.2 → aek_auto_mlbuilder-0.10.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aek-auto-mlbuilder",        
-    version="0.8.2",
+    version="0.10.0",
     description="Automatic ML model builder in Python",
     long_description=open("README.md", encoding="utf-8").read(),
     long_description_content_type="text/markdown",