alexkueck commited on
Commit
7094d26
·
1 Parent(s): 7dab25d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -365,8 +365,10 @@ print("trained!!!!!")
365
  #recall: Recall ist ein Maß dafür, wie viele der tatsächlich positiven Beispiele das Modell richtig klassifiziert hat (möglichst nahe 1)
366
  #precision: Maß dafür, wie viele der vom Modell als positiv klassifizierten Beispiele tatsächlich positiv sind (möglichst nahe 1)
367
  print ("#################Evaluate:#################")
368
- loss, accuracy, f1_score, recall = trainer.evaluate(eval_dataset=lm_datasets["test"])
369
- print(loss, accuracy, f1_score, recall)
 
 
370
  print("Done Eval")
371
 
372
  print('################ Test Trained Model ###################')
 
365
  #recall: Recall ist ein Maß dafür, wie viele der tatsächlich positiven Beispiele das Modell richtig klassifiziert hat (möglichst nahe 1)
366
  #precision: Maß dafür, wie viele der vom Modell als positiv klassifizierten Beispiele tatsächlich positiv sind (möglichst nahe 1)
367
  print ("#################Evaluate:#################")
368
+ results = trainer.evaluate(eval_dataset=lm_datasets["test"])
369
+ # Ausgabe der Ergebnisse
370
+ for key, value in results.items():
371
+ print(key, value)
372
  print("Done Eval")
373
 
374
  print('################ Test Trained Model ###################')