We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3e60674 commit 9afe24cCopy full SHA for 9afe24c
1 file changed
dataikuapi/dss/ml.py
@@ -1632,8 +1632,18 @@ def remove_sample_weighting(self):
1632
class DSSClusteringMLTaskSettings(DSSMLTaskSettings):
1633
__doc__ = []
1634
algorithm_remap = {
1635
- "DBSCAN" : "db_scan_clustering",
1636
- }
+ "DBSCAN": "db_scan_clustering",
+ "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
+ }
1647
1648
def get_algorithm_settings(self, algorithm_name):
1649
"""
0 commit comments