We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eaba956 commit 5226d24Copy full SHA for 5226d24
src/sasctl/tasks.py
@@ -494,9 +494,11 @@ def register_model(
494
model_prefix=name,
495
project=project,
496
input_data=info.X,
497
- predict_method=info.predict_function,
+ predict_method=[info.predict_function, info.y.iloc[0].to_list()],
498
predict_threshold=info.threshold,
499
- target_values=info.target_values)
+ score_metrics=info.output_column_names,
500
+ target_values=info.target_values,
501
+ model_file_name=list(pickled_model.keys())[0])
502
return model_obj
503
504
# # If the model is a scikit-learn model, generate the model dictionary
0 commit comments