Merge pull request #118 from dataiku/fix/clustering-mltask

Fix clustering mltask

Author: samronsin

dataikuapi/dss/ml.py

@@ -1429,8 +1429,9 @@ class PredictionTypes:
     def __init__(self, client, project_key, analysis_id, mltask_id, mltask_settings):
         DSSMLTaskSettings.__init__(self, client, project_key, analysis_id, mltask_id, mltask_settings)
 
-        if self.get_prediction_type() not in [self.PredictionTypes.BINARY, self.PredictionTypes.REGRESSION, self.PredictionTypes.MULTICLASS]:
-            raise ValueError("Unknown prediction type: {}".format(self.prediction_type))
+        prediction_type = self.get_prediction_type()
+        if prediction_type not in [self.PredictionTypes.BINARY, self.PredictionTypes.REGRESSION, self.PredictionTypes.MULTICLASS]:
+            raise ValueError("Unknown prediction type: {}".format(prediction_type))
 
         self.classification_prediction_types = [self.PredictionTypes.BINARY, self.PredictionTypes.MULTICLASS]
 
@@ -1632,8 +1633,18 @@ def remove_sample_weighting(self):
 class DSSClusteringMLTaskSettings(DSSMLTaskSettings):
     __doc__ = []
     algorithm_remap = {
-            "DBSCAN" : "db_scan_clustering",
-        }
+        "DBSCAN": "db_scan_clustering",
+        "SPECTRAL": "spectral_clustering",
+        "WARD": "ward_clustering",
+        "KMEANS": "kmeans_clustering",
+        "MINIBATCH_KMEANS": "mini_batch_kmeans_clustering",
+        "GAUSSIAN_MIXTURE": "gmm_clustering",
+        "TWO_STEP": "two_step",
+        "ISOLATION_FOREST": "isolation_forest",
+        "MLLIB_KMEANS": "mllib_kmeans_clustering",
+        "MLLIB_GAUSSIAN_MIXTURE": "mllib_gaussian_mixture_clustering",
+        "H20_KMEANS": "h2o_kmeans"
+    }
 
     def get_algorithm_settings(self, algorithm_name):
         """