diff --git a/example-get-started/code/src/evaluate.py b/example-get-started/code/src/evaluate.py index 599d73d..1c0d95f 100644 --- a/example-get-started/code/src/evaluate.py +++ b/example-get-started/code/src/evaluate.py @@ -100,7 +100,7 @@ def main(): test, _ = pickle.load(fd) # Evaluate train and test datasets. - with Live(EVAL_PATH, dvcyaml=False) as live: + with Live(EVAL_PATH) as live: evaluate(model, train, "train", live, save_path=EVAL_PATH) evaluate(model, test, "test", live, save_path=EVAL_PATH) diff --git a/example-get-started/generate.sh b/example-get-started/generate.sh index 41fa62f..339c57a 100755 --- a/example-get-started/generate.sh +++ b/example-get-started/generate.sh @@ -320,6 +320,9 @@ EOF python src/evaluate.py model.pkl data/features fi + sed -e "s/Live(EVAL_PATH)/Live(EVAL_PATH, dvcyaml=False)/" -i".bck" src/evaluate.py + rm -f src/evaluate.py.bck + echo "metrics: - eval/metrics.json plots: