Kevin Fink
commited on
Commit
·
c850f94
1
Parent(s):
2495767
deve
Browse files
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["
|
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]
|