| { | |
| "model_name": "sms-spam-detector", | |
| "base_model": "distilbert-base-uncased", | |
| "task": "text-classification", | |
| "language": "en", | |
| "dataset": "SMS Spam Collection", | |
| "metrics": { | |
| "accuracy": 0.9916267942583732, | |
| "precision": 0.972972972972973, | |
| "recall": 0.9642857142857143, | |
| "f1_score": 0.968609865470852, | |
| "roc_auc": 0.9990380820836622 | |
| }, | |
| "training_params": { | |
| "epochs": 3, | |
| "batch_size": 16, | |
| "learning_rate": 2e-05 | |
| } | |
| } |