aek-auto-mlbuilder 0.7.0__py3-none-any.whl → 0.8.1__py3-none-any.whl

aek_auto_mlbuilder/__init__.py

@@ -6,4 +6,5 @@ from .decision_tree import DecisionTreeModel
 from .knn import KNNModel
 from .random_forest import RandomForestModel
 from .svm import SVMModel
-from .naive_bayes import NaiveBayesModel
+from .naive_bayes import NaiveBayesModel
+from .kmeans import KMeansModel

aek_auto_mlbuilder/kmeans.py

@@ -0,0 +1,41 @@
+from sklearn.cluster import KMeans
+from .base import BaseModel
+
+
+class KMeansModel(BaseModel):
+    """
+    K-Means clustering model (unsupervised learning)
+    Brute force method has been using
+    """
+    def __init__(self, param_grid=None):
+        super().__init__()
+        self.param_grid = param_grid or {
+            "n_clusters": [2, 3, 4, 5, 6],
+            "init": ["k-means++", "random"],
+            "n_init": [10, 20],
+            "max_iter": [300, 500]
+        }
+
+    def train(self, X):
+        best_score = -float("inf")
+        best_model = None
+
+        for n_clusters in self.param_grid["n_clusters"]:
+            for init in self.param_grid["init"]:
+                for n_init in self.param_grid["n_init"]:
+                    for max_iter in self.param_grid["max_iter"]:
+                        model = KMeans(
+                            n_clusters=n_clusters,
+                            init=init,
+                            n_init=n_init,
+                            max_iter=max_iter,
+                            random_state=42
+                        )
+                        model.fit(X)
+                        score = -model.inertia_
+                        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/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 is using
+    Brute-force search has been using
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

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 is using
+    brute force hyperparameter search has been using
     """
     def __init__(self, nb_type="gaussian", param_grid=None):
         super().__init__()

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 is using
+    brute force method has been using
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

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 is using
+    brute force has been using
     """
     def __init__(self, task="classification", param_grid=None):
         super().__init__()

{aek_auto_mlbuilder-0.7.0.dist-info → aek_auto_mlbuilder-0.8.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.7.0
+Version: 0.8.1
 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.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+aek_auto_mlbuilder/__init__.py,sha256=1fgkK5aJ9Y-ig87L4puE4vGEjjy8oPdrZTeGo3GuI7Q,371
+aek_auto_mlbuilder/base.py,sha256=GgMdAoceRjwz3i9rVQ0RAjvn5ZdRS-sAkLWjymbE8s0,385
+aek_auto_mlbuilder/decision_tree.py,sha256=OImOHaREz2jWcANXVC6VKcatFJfzrtyCHJKXNA5-hoI,1606
+aek_auto_mlbuilder/kmeans.py,sha256=GM1sBV0HQoTjXSVob8v9-iAvUaGhu4UVlArxCgPHTXk,1413
+aek_auto_mlbuilder/knn.py,sha256=ImBpgCJHTFTC9VnPI5yt53pjY91eNxUKop2PPpELQRM,1728
+aek_auto_mlbuilder/linear_regression.py,sha256=MtOSRiXDIJPd3abnz4yNT4DBtrkmvEy00Kbx4AFk4Kg,1259
+aek_auto_mlbuilder/logistic_regression.py,sha256=lp9-e9p9QrqL20DmhIJaSnra1SwyMiOdTfOMlgYsNQA,1707
+aek_auto_mlbuilder/naive_bayes.py,sha256=3eWeDOwUJL5BwtKH0XvhqaS2c4nnXf8EqCxvEcuqXuY,2447
+aek_auto_mlbuilder/random_forest.py,sha256=HDTHh9JgxyKYfFzUeZug9-mVoNy5jQ0v7lVSigBRPR8,2024
+aek_auto_mlbuilder/svm.py,sha256=mMIT30Cc2JGMVat_2IcYCGWE3M23EcunWZNWcAL2ZG0,1620
+aek_auto_mlbuilder/utils.py,sha256=NcoM3b4Ng1Ogk3iKuz9DcMVwppGRqOLRp5g9jBCkWxY,190
+aek_auto_mlbuilder-0.8.1.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
+aek_auto_mlbuilder-0.8.1.dist-info/METADATA,sha256=mqMVQInD-zo04aubBUS4cXsQOavyxgGCljIAXw6JYlQ,1400
+aek_auto_mlbuilder-0.8.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aek_auto_mlbuilder-0.8.1.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
+aek_auto_mlbuilder-0.8.1.dist-info/RECORD,,

aek_auto_mlbuilder-0.7.0.dist-info/RECORD

@@ -1,15 +0,0 @@
-aek_auto_mlbuilder/__init__.py,sha256=6EsICSbltAGNZCv928oJGIBfGAB1Wc7swLHzFXnsqfY,339
-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/naive_bayes.py,sha256=FfnB2SOw-4ySWdQ4kLx1wTQkbftXypwZc5koZ5WqbOA,2441
-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.7.0.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
-aek_auto_mlbuilder-0.7.0.dist-info/METADATA,sha256=Nh8czVd0Ws6TUBLeRKmejv06v4r4G7vHPPKp7VrfeJg,1400
-aek_auto_mlbuilder-0.7.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aek_auto_mlbuilder-0.7.0.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
-aek_auto_mlbuilder-0.7.0.dist-info/RECORD,,