Update README.md
Browse files
README.md
CHANGED
@@ -15,16 +15,16 @@ should probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
# food_type_classification_model
|
17 |
|
18 |
-
Easily classify any food product into Plant-based(
|
19 |
|
20 |
-
This model was trained using a dataset from [USDA FoodData Central](https://fdc.nal.usda.gov/download-datasets.html) which contains
|
21 |
the ANIMAL_BASED and PLANT_BASED classification labels based on the available protein type in a food product.
|
22 |
|
23 |
|
24 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
|
25 |
It achieves the following results on the evaluation set:
|
26 |
-
- Loss: 0.0249
|
27 |
-
- Accuracy: 0.9940
|
28 |
|
29 |
## Model description
|
30 |
|
|
|
15 |
|
16 |
# food_type_classification_model
|
17 |
|
18 |
+
Easily classify any food product into Plant-based(**PLANT_BASED**) or Animal-based(**ANIMAL_BASED**) based on ingredients or product title.
|
19 |
|
20 |
+
This model was trained using a dataset from **[USDA FoodData Central](https://fdc.nal.usda.gov/download-datasets.html)** which contains
|
21 |
the ANIMAL_BASED and PLANT_BASED classification labels based on the available protein type in a food product.
|
22 |
|
23 |
|
24 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
|
25 |
It achieves the following results on the evaluation set:
|
26 |
+
- **Loss: 0.0249**
|
27 |
+
- **Accuracy: 0.9940**
|
28 |
|
29 |
## Model description
|
30 |
|