Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save AntoineToubhans/59b8c58b5bc7550926e29552d2f7b567 to your computer and use it in GitHub Desktop.
Save AntoineToubhans/59b8c58b5bc7550926e29552d2f7b567 to your computer and use it in GitHub Desktop.
stages:
download_dataset: ...
split_dataset: ...
train: ...
evaluate:
cmd: python scripts/evaluate.py
deps:
- scripts/evaluate.py
- data/dataset/test
- data/train/model
metrics:
- data/evaluation/metrics.json:
cache: false
plots:
- data/evaluation/predictions.csv:
template: confusion
x: true_label
y: predicted_label
title: Confusion Matrix
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment