| { |
| "validation": { |
| "eval_loss": 0.23355363309383392, |
| "eval_accuracy": 0.9453333333333334, |
| "eval_precision_macro": 0.9456396536674079, |
| "eval_recall_macro": 0.9453333333333332, |
| "eval_f1_macro": 0.9453725098595114, |
| "eval_runtime": 17.872, |
| "eval_samples_per_second": 671.442, |
| "eval_steps_per_second": 20.983, |
| "epoch": 3.0 |
| }, |
| "test": { |
| "eval_loss": 0.23390814661979675, |
| "eval_accuracy": 0.9460526315789474, |
| "eval_precision_macro": 0.9461382514079757, |
| "eval_recall_macro": 0.9460526315789474, |
| "eval_f1_macro": 0.9460740659250028, |
| "eval_runtime": 11.2019, |
| "eval_samples_per_second": 678.456, |
| "eval_steps_per_second": 21.246, |
| "epoch": 3.0 |
| }, |
| "classification_report": { |
| "World": { |
| "precision": 0.9602964531498147, |
| "recall": 0.9547368421052631, |
| "f1-score": 0.9575085774610715, |
| "support": 1900.0 |
| }, |
| "Sports": { |
| "precision": 0.9884149552395998, |
| "recall": 0.9878947368421053, |
| "f1-score": 0.9881547775730455, |
| "support": 1900.0 |
| }, |
| "Business": { |
| "precision": 0.9202975557917109, |
| "recall": 0.911578947368421, |
| "f1-score": 0.9159175039661555, |
| "support": 1900.0 |
| }, |
| "Sci/Tech": { |
| "precision": 0.9155440414507772, |
| "recall": 0.93, |
| "f1-score": 0.9227154046997389, |
| "support": 1900.0 |
| }, |
| "accuracy": 0.9460526315789474, |
| "macro avg": { |
| "precision": 0.9461382514079757, |
| "recall": 0.9460526315789474, |
| "f1-score": 0.9460740659250028, |
| "support": 7600.0 |
| }, |
| "weighted avg": { |
| "precision": 0.9461382514079756, |
| "recall": 0.9460526315789474, |
| "f1-score": 0.9460740659250028, |
| "support": 7600.0 |
| } |
| }, |
| "labels": { |
| "0": "World", |
| "1": "Sports", |
| "2": "Business", |
| "3": "Sci/Tech" |
| }, |
| "training_config": { |
| "base_model": "google-bert/bert-base-uncased", |
| "dataset": "fancyzhx/ag_news", |
| "max_length": 256, |
| "train_batch_size": 16, |
| "eval_batch_size": 32, |
| "learning_rate": 2e-05, |
| "num_epochs": 3.0, |
| "weight_decay": 0.01, |
| "warmup_ratio": 0.1 |
| } |
| } |
