sdtemple
/

shape-prediction-model

Image Classification

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

sdtemple commited on 3 days ago

Commit

a745d82

·

verified ·

1 Parent(s): 9f8b5de

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -20,7 +20,7 @@ This model is a part of a how to tutorial on fitting PyTorch models.
 The model is trained on 2000 examples for each color and shape combo (64,000 samples in total) simulated according to [https://github.com/sdtemple/zootopia3](https://github.com/sdtemple/zootopia3).
-The model is tested/evaluated on the dataset [https://huggingface.co/datasets/sdtemple/colored-shapes](https://huggingface.co/datasets/sdtemple/colored-shapes) which has slightly smaller shapes simulated (out of distribution) relative to the training data.
 - Accuracy: 75%
 - Min precision (triangle): 57%

 The model is trained on 2000 examples for each color and shape combo (64,000 samples in total) simulated according to [https://github.com/sdtemple/zootopia3](https://github.com/sdtemple/zootopia3).
+The model is tested/evaluated on the dataset [https://huggingface.co/datasets/sdtemple/colored-shapes](https://huggingface.co/datasets/sdtemple/colored-shapes), which has slightly smaller shapes simulated (out of distribution) relative to the training data. The metrics below can be +- a few points depending on random seed.
 - Accuracy: 75%
 - Min precision (triangle): 57%