Update README.md

README.md

@@ -23,13 +23,13 @@ The model is trained on 2000 examples for each color and shape combo (64,000 sam
 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): 56%
-- Max precision (rectangle): 89%
-- Min recall (diamond): 59%
-- Max recall (triangle): 86%
-- AUROC (macro-averaged): 91%
+- Min precision (triangle): 57%
+- Max precision (rectangle): 98%
+- Min recall (diamond): 66%
+- Max recall (triangle): 84%
+- AUROC (macro-averaged): 92%
 - Min AUROC (diamond): 90%
-- Max AUROC (circle): 93%
+- Max AUROC (circle): 94%
 
 Compared to [https://huggingface.co/sdtemple/color-prediction-model](https://huggingface.co/sdtemple/color-prediction-model), it is harder to predict the shape than the color of the object.