aek-auto-mlbuilder 0.9.0__py3-none-any.whl → 0.10.1__py3-none-any.whl

aek_auto_mlbuilder/__init__.py

@@ -8,4 +8,5 @@ from .random_forest import RandomForestModel
 from .svm import SVMModel
 from .naive_bayes import NaiveBayesModel
 from .kmeans import KMeansModel
-from .dbscan import DBSCANModel
+from .dbscan import DBSCANModel
+from .agglomerative import AgglomerativeModel

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.9.0.dist-info → aek_auto_mlbuilder-0.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aek-auto-mlbuilder
-Version: 0.9.0
+Version: 0.10.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.9.0.dist-info → aek_auto_mlbuilder-0.10.1.dist-info}/RECORD

@@ -1,4 +1,5 @@
-aek_auto_mlbuilder/__init__.py,sha256=tkmthhNIRQ3pidoz43dBPgeHcrSjxWANyil6QYvojKY,403
+aek_auto_mlbuilder/__init__.py,sha256=7CehkJhvgV-aXCeAOHTRQ02XfpmzxQ3M4QgluGBPysA,449
+aek_auto_mlbuilder/agglomerative.py,sha256=AK_Vu6werEn64WymqAxTe_E2CLS4ELnGOSlqVCUAcqY,1262
 aek_auto_mlbuilder/base.py,sha256=GgMdAoceRjwz3i9rVQ0RAjvn5ZdRS-sAkLWjymbE8s0,385
 aek_auto_mlbuilder/dbscan.py,sha256=Z9jXoixn7L2WFOHChkWaiZ9sj_Qo0j5R7_kNMacL0ik,986
 aek_auto_mlbuilder/decision_tree.py,sha256=OImOHaREz2jWcANXVC6VKcatFJfzrtyCHJKXNA5-hoI,1606
@@ -10,8 +11,8 @@ aek_auto_mlbuilder/naive_bayes.py,sha256=UCZrLOTo7ibgwSQH5aqZgyrDvMWkM4-aQwQJDk_
 aek_auto_mlbuilder/random_forest.py,sha256=RpT5-QX1D6iGjRtcPFuRtqMrdmnDNklA9w3uwbtYYlM,2023
 aek_auto_mlbuilder/svm.py,sha256=-IQjzHIzHhgpi2mIgWVzIDKlhAS7o_tMVQhN_epseKw,1619
 aek_auto_mlbuilder/utils.py,sha256=NcoM3b4Ng1Ogk3iKuz9DcMVwppGRqOLRp5g9jBCkWxY,190
-aek_auto_mlbuilder-0.9.0.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
-aek_auto_mlbuilder-0.9.0.dist-info/METADATA,sha256=LFwhjfjzCogh_2nbW3lM7h68t3UnQc24HeGBDNBJBZo,1400
-aek_auto_mlbuilder-0.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aek_auto_mlbuilder-0.9.0.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
-aek_auto_mlbuilder-0.9.0.dist-info/RECORD,,
+aek_auto_mlbuilder-0.10.1.dist-info/licenses/LICENSE,sha256=eSVo2jJj9FB1xvr0zZ9U1fXkyjjnT6-WM3O4HSFKJOc,133
+aek_auto_mlbuilder-0.10.1.dist-info/METADATA,sha256=-GfY2_xSPv5DzUuWWuq3mWUrm28oEQUa-xYmljHiF2Y,1401
+aek_auto_mlbuilder-0.10.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aek_auto_mlbuilder-0.10.1.dist-info/top_level.txt,sha256=2ZY5rMRnVvrAH2GRGUbd6n9ey8cg_uk5iJwke0hQzFE,19
+aek_auto_mlbuilder-0.10.1.dist-info/RECORD,,