Skip to content

Commit 99fc47c

Browse files
author
aavtomonov
committed
pass reguess level explicitly
1 parent 6665cfc commit 99fc47c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dataikuapi/dss/ml.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1686,18 +1686,18 @@ 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, target_only=None):
1689+
def guess(self, prediction_type=None, reguess_level=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 target_only: if True then skip ML task settings automatic detection, only the target values are remapped. Only valid for prediction ML Tasks.
1693+
:param bool reguess_level: One of the following values: TARGET_CHANGE, TARGET_REGUESS and FULL_REGUESS. Only valid for prediction ML Tasks.
16941694
"""
16951695
obj = {}
16961696
if prediction_type is not None:
16971697
obj["predictionType"] = prediction_type
16981698

1699-
if target_only is not None:
1700-
obj["targetOnly"] = target_only
1699+
if reguess_level is not None:
1700+
obj["reguessLevel"] = reguess_level
17011701

17021702
self.client._perform_empty(
17031703
"PUT",

0 commit comments

Comments
 (0)