alexkueck commited on
Commit
844fcc5
·
1 Parent(s): 81bd4fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def group_texts(examples):
40
  #Funktion, die der trainer braucht, um das Training zu evaluieren - mit einer Metrik
41
  def compute_metrics(eval_pred):
42
  #Metrik berechnen, um das training messen zu können - wird es besser???
43
- metric = evaluate.load("accuracy")
44
  logits, labels = eval_pred
45
  predictions = np.argmax(logits, axis=-1)
46
  #Call compute on metric to calculate the accuracy of your predictions.
 
40
  #Funktion, die der trainer braucht, um das Training zu evaluieren - mit einer Metrik
41
  def compute_metrics(eval_pred):
42
  #Metrik berechnen, um das training messen zu können - wird es besser???
43
+ metric = evaluate.load("accuracy") #3 Arten von gegebener Metrik: f1 oder roc_auc oder accuracy
44
  logits, labels = eval_pred
45
  predictions = np.argmax(logits, axis=-1)
46
  #Call compute on metric to calculate the accuracy of your predictions.