Update README.md
Browse files
README.md
CHANGED
|
@@ -19,17 +19,25 @@ We used the whole Recipe1M+ dataset with a total of 1,029,720 records, with usin
|
|
| 19 |
It achieves the following results on the evaluation set:
|
| 20 |
- Loss: 0.6230
|
| 21 |
|
| 22 |
-
##
|
| 23 |
|
| 24 |
-
|
| 25 |
|
| 26 |
-
|
|
|
|
| 27 |
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
-
## Training and evaluation data
|
| 31 |
|
| 32 |
-
More information needed
|
| 33 |
|
| 34 |
## Training procedure
|
| 35 |
|
|
|
|
| 19 |
It achieves the following results on the evaluation set:
|
| 20 |
- Loss: 0.6230
|
| 21 |
|
| 22 |
+
## Usage
|
| 23 |
|
| 24 |
+
You can use this model to get embeddings/representations for your food-related dataset that you will use if for your downstream tasks.
|
| 25 |
|
| 26 |
+
```python
|
| 27 |
+
from transformers import pipeline
|
| 28 |
|
| 29 |
+
# Your food-related data
|
| 30 |
+
food_data = "Hawaiian Pizza"
|
| 31 |
+
# Use pipeline for feature extraction
|
| 32 |
+
embedding = pipeline(
|
| 33 |
+
'feature-extraction', model='alexdseo/RecipeBERT', framework='pt'
|
| 34 |
+
)
|
| 35 |
+
# Meal pooling
|
| 36 |
+
food_rep = embedding(food_data, return_tensors='pt')[0].numpy().mean(axis=0)
|
| 37 |
+
|
| 38 |
+
```
|
| 39 |
|
|
|
|
| 40 |
|
|
|
|
| 41 |
|
| 42 |
## Training procedure
|
| 43 |
|