Rezky Mulia Kam commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,6 +11,9 @@ tags:
|
|
| 11 |
- finetune
|
| 12 |
- paulekman
|
| 13 |
---
|
|
|
|
|
|
|
|
|
|
| 14 |
final_training_args = TrainingArguments(
|
| 15 |
output_dir=outputs_dir,
|
| 16 |
per_device_train_batch_size=32,
|
|
|
|
| 11 |
- finetune
|
| 12 |
- paulekman
|
| 13 |
---
|
| 14 |
+
|
| 15 |
+
label_map = {0: 'sadness', 1: 'joy', 2: 'love', 3: 'anger', 4: 'fear', 5: 'surprise'}
|
| 16 |
+
|
| 17 |
final_training_args = TrainingArguments(
|
| 18 |
output_dir=outputs_dir,
|
| 19 |
per_device_train_batch_size=32,
|