From 3eb148c15cb2835a9c7540f44b1c191f111d87b7 Mon Sep 17 00:00:00 2001 From: Dave Berenbaum Date: Fri, 19 Apr 2024 13:46:36 -0400 Subject: [PATCH] use default dvclive plots (#282) --- example-get-started/code/src/evaluate.py | 2 +- example-get-started/generate.sh | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) 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: