Merge pull request #27 from dataiku/feature/dss50-analysis-cleanup

Analysis cleanup API

Author: cstenac

dataikuapi/dss/ml.py

@@ -1672,6 +1672,15 @@ def get_trained_model_details(self, id):
         else:
             return DSSTrainedPredictionModelDetails(ret, snippet, mltask=self, mltask_model_id=id)
 
+    def delete_trained_model(self, model_id):
+        """
+        Deletes a trained model
+
+        :param str model_id: Model identifier, as returend by :meth:`get_trained_models_ids`
+        """
+        self.client._perform_empty(
+            "DELETE", "/projects/%s/models/lab/%s/%s/models/%s" % (self.project_key, self.analysis_id, self.mltask_id, model_id))
+
     def deploy_to_flow(self, model_id, model_name, train_dataset, test_dataset=None, redo_optimization=True):
         """
         Deploys a trained model from this ML Task to a saved model + train recipe in the Flow.
@@ -1716,6 +1725,29 @@ def redeploy_to_flow(self, model_id, recipe_name=None, saved_model_id=None, acti
             "POST", "/projects/%s/models/lab/%s/%s/models/%s/actions/redeployToFlow" % (self.project_key, self.analysis_id, self.mltask_id, model_id),
             body = obj)
 
+    def remove_unused_splits(self):
+        """
+        Deletes all stored splits data that are not anymore in use for this ML Task.
+
+        It is generally not needed to call this method
+        """
+        self.client._perform_empty(
+            "POST", "/projects/%s/models/lab/%s/%s/actions/removeUnusedSplits" % (self.project_key, self.analysis_id, self.mltask_id))
+
+    def remove_all_splits(self):
+        """
+        Deletes all stored splits data for this ML Task. This operation saves disk space.
+
+        After performing this operation, it will not be possible anymore to:
+        * Ensemble already trained models
+        * View the "predicted data" or "charts" for already trained models
+        * Resume training of models for which optimization had been previously interrupted
+        
+        Training new models remains possible
+        """
+        self.client._perform_empty(
+            "POST", "/projects/%s/models/lab/%s/%s/actions/removeAllSplits" % (self.project_key, self.analysis_id, self.mltask_id))
+
     def guess(self, prediction_type=None, reguess_level=None):
         """
         Guess the feature handling and the algorithms.