| { | |
| "model_name": "roberta-base", | |
| "training_samples": 1629193, | |
| "validation_samples": 325839, | |
| "test_samples": 217226, | |
| "epochs": 5, | |
| "batch_size": 24, | |
| "learning_rate": 2e-05, | |
| "max_length": 256, | |
| "validation_metrics": { | |
| "eval_loss": 0.148252934217453, | |
| "eval_accuracy": 0.9454884160582373, | |
| "eval_f1": 0.9250213175513099, | |
| "eval_precision": 0.9245607817325705, | |
| "eval_recall": 0.9254823123965267, | |
| "eval_roc_auc": 0.9887611402018297, | |
| "eval_true_positives": 109566.0, | |
| "eval_true_negatives": 198511.0, | |
| "eval_false_positives": 8940.0, | |
| "eval_false_negatives": 8822.0, | |
| "eval_runtime": 195.9861, | |
| "eval_samples_per_second": 1662.561, | |
| "eval_steps_per_second": 34.64, | |
| "epoch": 5.0 | |
| }, | |
| "test_metrics": { | |
| "eval_loss": 0.148285910487175, | |
| "eval_accuracy": 0.9463231841492271, | |
| "eval_f1": 0.9261520533022572, | |
| "eval_precision": 0.9259057581014855, | |
| "eval_recall": 0.9263984795692113, | |
| "eval_roc_auc": 0.9889512279429393, | |
| "eval_true_positives": 73116.0, | |
| "eval_true_negatives": 132450.0, | |
| "eval_false_positives": 5851.0, | |
| "eval_false_negatives": 5809.0, | |
| "eval_runtime": 142.5284, | |
| "eval_samples_per_second": 1524.09, | |
| "eval_steps_per_second": 31.755, | |
| "epoch": 5.0 | |
| } | |
| } |