Skip to content

Commit f4308e0

Browse files
author
aavtomonov
committed
Added redetect flag to ML task guess public API
1 parent 1f51e1e commit f4308e0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

dataikuapi/dss/ml.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1686,15 +1686,16 @@ def redeploy_to_flow(self, model_id, recipe_name=None, saved_model_id=None, acti
16861686
"POST", "/projects/%s/models/lab/%s/%s/models/%s/actions/redeployToFlow" % (self.project_key, self.analysis_id, self.mltask_id, model_id),
16871687
body = obj)
16881688

1689-
def guess(self, prediction_type=None):
1689+
def guess(self, prediction_type=None, redetect=None):
16901690
"""
16911691
Guess the feature handling and the algorithms.
16921692
:param string prediction_type: In case of a prediction problem the prediction type can be specify. Valid values are BINARY_CLASSIFICATION, REGRESSION, MULTICLASS.
16931693
"""
16941694
obj = {}
16951695
if prediction_type is not None:
16961696
obj["predictionType"] = prediction_type
1697-
1697+
if redetect is not None:
1698+
obj["redetect"] = redetect
16981699
self.client._perform_empty(
16991700
"PUT",
17001701
"/projects/%s/models/lab/%s/%s/guess" % (self.project_key, self.analysis_id, self.mltask_id),

0 commit comments

Comments
 (0)