Skip to content

Commit 459d1c7

Browse files
committed
probabilistic output in create_model
1 parent 8dd09e6 commit 459d1c7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

notebooks/1_task_and_model.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,8 @@
289289
" name=\"model-name\",\n",
290290
" version=\"v0.2.1\",\n",
291291
" metric_name=ml3_enums.ModelMetricName.RMSE,\n",
292-
" preferred_suggestion_type=ml3_enums.SuggestionType.SAMPLE_WEIGHTS\n",
292+
" preferred_suggestion_type=ml3_enums.SuggestionType.SAMPLE_WEIGHTS,\n",
293+
" with_probabilistic_output=False,\n",
293294
")\n",
294295
"logger.info(f\"Created model with id {model_id}\")"
295296
]

0 commit comments

Comments
 (0)