Update README.md

README.md

@@ -12,21 +12,27 @@ probably proofread and complete it, then remove this comment. -->
 
 # distilbert-ft-test3
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
-It achieves the following results on the evaluation set:
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on [thomasavare/waste-classification-v2](https://huggingface.co/datasets/thomasavare/waste-classification-v2).
+It is part of my master thesis at Politecnico di Torino in partenership with ReLearn.
+
+It achieves the following results on the test set:
+
+accuracy | precision | recall | f1     |
+---------|-----------|--------|--------|
+0.974    | 0.9805    | 0.9732 | 0.9725 |
 
 
 ## Model description
 
-More information needed
+DistilBERT finetuned for waste classification on 50 different classes as part of my master thesis at Politecnico di Torino.
 
 ## Intended uses & limitations
 
-More information needed
+Use for waste classification on 50 different waste classes (see [dataset](https://huggingface.co/datasets/thomasavare/waste-classification-v2))
 
 ## Training and evaluation data
 
-More information needed
+[waste-classification-v2 dataset](https://huggingface.co/datasets/thomasavare/waste-classification-v2)
 
 ## Training procedure