Spaces:
Paused
Paused
Update app.py
Browse files
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.
|