han7ter commited on
Commit
7e3a23c
·
1 Parent(s): adf848c

FUCKING ERROR

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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
- logits, labels = eval_pred
50
- predictions = np.argmax(logits, axis=-1)
51
- return metric.compute(predictions=predictions, references=labels)
 
 
 
 
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(