mariadg commited on
Commit
04d2120
·
verified ·
1 Parent(s): 6b95560

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -62,14 +62,14 @@ print(pred) # Map this index back to label if needed
62
 
63
  The model outputs a numerical label corresponding to a prompt category. Below is the mapping between label IDs and their respective categories:
64
 
65
- 0: brainstorming
66
- 1: classification
67
- 2: closed_qa
68
- 3: creative_writing
69
- 4: general_qa
70
- 5: information_extraction
71
- 6: open_qa
72
- 7: summarization
73
 
74
  ## 🛠️ Training Details
75
 
 
62
 
63
  The model outputs a numerical label corresponding to a prompt category. Below is the mapping between label IDs and their respective categories:
64
 
65
+ 0: `brainstorming`
66
+ 1: `classification`
67
+ 2: `closed_qa`
68
+ 3: `creative_writing`
69
+ 4: `general_qa`
70
+ 5: `information_extraction`
71
+ 6: `open_qa`
72
+ 7: `summarization`
73
 
74
  ## 🛠️ Training Details
75