Update README.md
Browse files
README.md
CHANGED
|
@@ -62,13 +62,13 @@ 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
|
|
|
|
| 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
|