Spaces:
Runtime error
Runtime error
Create config.py
Browse files
config.py
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
model_name = ""
|
| 2 |
+
output_path = ""
|
| 3 |
+
drop_out = 0.3
|
| 4 |
+
max_length = 512
|
| 5 |
+
epochs = 5
|
| 6 |
+
label2id = {'neutral': 0, 'negative': 1, 'positive': 2}
|
| 7 |
+
id2label = {0: 'neutral', 1: 'negative', 2: 'positive'}
|
| 8 |
+
output_dir=""
|
| 9 |
+
batch_size=16 #32
|
| 10 |
+
learning_rate=2e-5
|