bert-tiny-amd / model_metadata.json
Adya662's picture
Upload model artifacts and README
e14b87b verified
{
"model_name": "prajjwal1/bert-tiny",
"max_length": 128,
"optimal_threshold": 0.11530892550945282,
"val_metrics": {
"accuracy": 0.9702797202797203,
"precision": 0.9004739336492891,
"recall": 0.979381443298969,
"f1": 0.9382716049382717
},
"test_metrics": {
"accuracy": 0.8353344768439108,
"precision": 0.6678445229681979,
"recall": 0.9895287958115183,
"f1": 0.7974683544303798,
"confusion_matrix": [
[
298,
94
],
[
2,
189
]
]
},
"dropout_rate": 0.2,
"training_date": "2025-10-04 02:43:10",
"hidden_size": 128
}