| { | |
| "model_architecture": "DistilBertForTokenClassification", | |
| "base_model": "distilbert-base-multilingual-cased", | |
| "epochs": 3, | |
| "best_val_macro_f1": 0.9931, | |
| "total_training_time_seconds": 124.26, | |
| "history": [ | |
| { | |
| "epoch": 1, | |
| "train_loss": 0.8143, | |
| "val_loss": 0.0702, | |
| "macro_f1": 0.9762, | |
| "categories": { | |
| "PERSON": { | |
| "precision": 0.9723, | |
| "recall": 0.9992, | |
| "f1": 0.9856, | |
| "support": 1299 | |
| }, | |
| "ADDRESS": { | |
| "precision": 0.9962, | |
| "recall": 1.0, | |
| "f1": 0.9981, | |
| "support": 2073 | |
| }, | |
| "ORGANIZATION": { | |
| "precision": 0.9983, | |
| "recall": 0.9598, | |
| "f1": 0.9787, | |
| "support": 622 | |
| }, | |
| "LOCATION": { | |
| "precision": 0.8962, | |
| "recall": 0.9939, | |
| "f1": 0.9425, | |
| "support": 165 | |
| } | |
| } | |
| }, | |
| { | |
| "epoch": 2, | |
| "train_loss": 0.0131, | |
| "val_loss": 0.0251, | |
| "macro_f1": 0.9931, | |
| "categories": { | |
| "PERSON": { | |
| "precision": 0.9738, | |
| "recall": 1.0, | |
| "f1": 0.9867, | |
| "support": 1299 | |
| }, | |
| "ADDRESS": { | |
| "precision": 0.9942, | |
| "recall": 1.0, | |
| "f1": 0.9971, | |
| "support": 2073 | |
| }, | |
| "ORGANIZATION": { | |
| "precision": 1.0, | |
| "recall": 0.9952, | |
| "f1": 0.9976, | |
| "support": 622 | |
| }, | |
| "LOCATION": { | |
| "precision": 0.9821, | |
| "recall": 1.0, | |
| "f1": 0.991, | |
| "support": 165 | |
| } | |
| } | |
| }, | |
| { | |
| "epoch": 3, | |
| "train_loss": 0.0036, | |
| "val_loss": 0.0259, | |
| "macro_f1": 0.9926, | |
| "categories": { | |
| "PERSON": { | |
| "precision": 0.9738, | |
| "recall": 1.0, | |
| "f1": 0.9867, | |
| "support": 1299 | |
| }, | |
| "ADDRESS": { | |
| "precision": 0.9976, | |
| "recall": 1.0, | |
| "f1": 0.9988, | |
| "support": 2073 | |
| }, | |
| "ORGANIZATION": { | |
| "precision": 1.0, | |
| "recall": 0.9936, | |
| "f1": 0.9968, | |
| "support": 622 | |
| }, | |
| "LOCATION": { | |
| "precision": 0.9763, | |
| "recall": 1.0, | |
| "f1": 0.988, | |
| "support": 165 | |
| } | |
| } | |
| } | |
| ], | |
| "labels": [ | |
| "O", | |
| "B-PER", | |
| "I-PER", | |
| "B-ADDR", | |
| "I-ADDR", | |
| "B-ORG", | |
| "I-ORG", | |
| "B-LOC", | |
| "I-LOC" | |
| ] | |
| } |