alexdseo commited on
Commit
11eb05a
·
1 Parent(s): 9bb5db0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -6
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
- ## Model description
23
 
24
- More information needed
25
 
26
- ## Intended uses & limitations
 
27
 
28
- More information needed
 
 
 
 
 
 
 
 
 
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