Skip to content

Commit ce9b441

Browse files
authored
Feature/dss100 sc 79583 drift ser create a ser with a reference dataset (#218)
* Added optional input to Standalone Evaluation Recipe * Added documentation of new optional input of Standalone Evaluation Recipe * Modified docstring of optional SER input
1 parent 0c9c5c0 commit ce9b441

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

dataikuapi/dss/recipe.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1411,6 +1411,12 @@ class StandaloneEvaluationRecipeCreator(DSSRecipeCreator):
14111411
builder.with_input("scored_dataset_to_evaluate")
14121412
builder.with_output_evaluation_store(evaluation_store_id)
14131413
1414+
# Add a reference dataset (optional) to compute data drift
1415+
1416+
builder.with_reference_dataset("reference_dataset")
1417+
1418+
# Finish creation of the recipe
1419+
14141420
new_recipe = builder.create()
14151421
14161422
# Modify the model parameters in the SER settings
@@ -1465,6 +1471,10 @@ def with_output_evaluation_store(self, mes_id):
14651471
"""Sets the output model evaluation store"""
14661472
return self._with_output(mes_id, role="main")
14671473

1474+
def with_reference_dataset(self, dataset_name):
1475+
"""Sets the dataset to use as a reference in data drift computation (optional)."""
1476+
return self._with_input(dataset_name, self.project.project_key, role="reference")
1477+
14681478

14691479
class ClusteringScoringRecipeCreator(SingleOutputRecipeCreator):
14701480
"""

0 commit comments

Comments
 (0)