dineth554's picture
Rename model_config_v2.json to model_config.json
8eb5a59 verified
{
"model_type": "LargeSinhalaClassifier",
"architecture": {
"input_dim": 240,
"hidden_dims": [
320,
256,
128,
64
],
"dropout": [
0.35,
0.35,
0.3,
0.3
],
"batch_norm": true,
"num_classes": 1
},
"total_parameters": 202049,
"performance_metrics": {
"accuracy": 1.0,
"precision": 1.0,
"recall": 1.0,
"f1_score": 1.0,
"confusion_matrix": [
[
2298
]
]
},
"training_config": {
"batch_size": 64,
"optimizer": "Adam",
"learning_rate": 0.001,
"weight_decay": 0.0001,
"scheduler": "ReduceLROnPlateau",
"epochs_trained": 50
},
"safetensor_file": "sinhala_language_model_v2.safetensors",
"input_features": "MFCC_statistics",
"feature_dimension": 240
}