Kevin Fink commited on
Commit
c850f94
·
1 Parent(s): 2495767
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
52
 
53
  # Calculate accuracy
54
  accuracy = accuracy_score(decoded_labels, decoded_preds)
55
- result["accuracy"] = round(accuracy * 100, 4)
56
 
57
  # Calculate average generation length
58
  prediction_lens = [np.count_nonzero(pred != tokenizer.pad_token_id) for pred in preds]
 
52
 
53
  # Calculate accuracy
54
  accuracy = accuracy_score(decoded_labels, decoded_preds)
55
+ result["eval_accuracy"] = round(accuracy * 100, 4)
56
 
57
  # Calculate average generation length
58
  prediction_lens = [np.count_nonzero(pred != tokenizer.pad_token_id) for pred in preds]