Skip to content

Commit 6665cfc

Browse files
author
aavtomonov
committed
renamed parameter to "target_only". Call reguessWithTarget with redetect=false to implement the logic
1 parent 0d1eea4 commit 6665cfc

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

dataikuapi/dss/ml.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1686,17 +1686,19 @@ 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, redetect=None):
1689+
def guess(self, prediction_type=None, target_only=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.
1693-
:param bool redetect: if False then skip ML task settings automatic detection, only the target values are remapped
1693+
:param bool target_only: if True then skip ML task settings automatic detection, only the target values are remapped. Only valid for prediction ML Tasks.
16941694
"""
16951695
obj = {}
16961696
if prediction_type is not None:
16971697
obj["predictionType"] = prediction_type
1698-
if redetect is not None:
1699-
obj["redetect"] = redetect
1698+
1699+
if target_only is not None:
1700+
obj["targetOnly"] = target_only
1701+
17001702
self.client._perform_empty(
17011703
"PUT",
17021704
"/projects/%s/models/lab/%s/%s/guess" % (self.project_key, self.analysis_id, self.mltask_id),

0 commit comments

Comments
 (0)