Skip to content

Commit

Permalink
use default dvclive plots (#282)
Browse files Browse the repository at this point in the history
  • Loading branch information
dberenbaum committed Apr 19, 2024
1 parent bc47159 commit 3eb148c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
2 changes: 1 addition & 1 deletion example-get-started/code/src/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down
3 changes: 3 additions & 0 deletions example-get-started/generate.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 3eb148c

Please sign in to comment.