Skip to content

Commit 5226d24

Browse files
committed
feat: generate score code
1 parent eaba956 commit 5226d24

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sasctl/tasks.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,9 +494,11 @@ def register_model(
494494
model_prefix=name,
495495
project=project,
496496
input_data=info.X,
497-
predict_method=info.predict_function,
497+
predict_method=[info.predict_function, info.y.iloc[0].to_list()],
498498
predict_threshold=info.threshold,
499-
target_values=info.target_values)
499+
score_metrics=info.output_column_names,
500+
target_values=info.target_values,
501+
model_file_name=list(pickled_model.keys())[0])
500502
return model_obj
501503

502504
# # If the model is a scikit-learn model, generate the model dictionary

0 commit comments

Comments
 (0)