Spaces:
Runtime error
Runtime error
FUCKING ERROR
Browse files
app.py
CHANGED
|
@@ -43,12 +43,15 @@ training_args = TrainingArguments(
|
|
| 43 |
report_to='none'
|
| 44 |
)
|
| 45 |
|
| 46 |
-
# Определяем как считать метрику
|
| 47 |
metric = evaluate.load('f1')
|
| 48 |
def compute_metrics(eval_pred):
|
| 49 |
-
|
| 50 |
-
|
| 51 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
# Выполняем обучение
|
| 54 |
trainer = Trainer(
|
|
|
|
| 43 |
report_to='none'
|
| 44 |
)
|
| 45 |
|
|
|
|
| 46 |
metric = evaluate.load('f1')
|
| 47 |
def compute_metrics(eval_pred):
|
| 48 |
+
logits, labels = eval_pred
|
| 49 |
+
predictions = np.argmax(logits, axis=-1)
|
| 50 |
+
return metric.compute(
|
| 51 |
+
predictions=predictions,
|
| 52 |
+
references=labels,
|
| 53 |
+
average='micro'
|
| 54 |
+
)
|
| 55 |
|
| 56 |
# Выполняем обучение
|
| 57 |
trainer = Trainer(
|