Skip to content

Commit 9afe24c

Browse files
committed
Complete algorithm_remap for DSSClusteringMLTaskSettings
1 parent 3e60674 commit 9afe24c

1 file changed

Lines changed: 12 additions & 2 deletions

File tree

dataikuapi/dss/ml.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1632,8 +1632,18 @@ def remove_sample_weighting(self):
16321632
class DSSClusteringMLTaskSettings(DSSMLTaskSettings):
16331633
__doc__ = []
16341634
algorithm_remap = {
1635-
"DBSCAN" : "db_scan_clustering",
1636-
}
1635+
"DBSCAN": "db_scan_clustering",
1636+
"SPECTRAL": "spectral_clustering",
1637+
"WARD": "ward_clustering",
1638+
"KMEANS": "kmeans_clustering",
1639+
"MINIBATCH_KMEANS": "mini_batch_kmeans_clustering",
1640+
"GAUSSIAN_MIXTURE": "gmm_clustering",
1641+
"TWO_STEP": "two_step",
1642+
"ISOLATION_FOREST": "isolation_forest",
1643+
"MLLIB_KMEANS": "mllib_kmeans_clustering",
1644+
"MLLIB_GAUSSIAN_MIXTURE": "mllib_gaussian_mixture_clustering",
1645+
"H20_KMEANS": "h2o_kmeans"
1646+
}
16371647

16381648
def get_algorithm_settings(self, algorithm_name):
16391649
"""

0 commit comments

Comments
 (0)