| { | |
| "model_type": "CompactGenderClassifier", | |
| "architectures": [ | |
| "CompactGenderClassifier" | |
| ], | |
| "vocab_size": 4000, | |
| "emb_dim": 64, | |
| "hidden_dim": 128, | |
| "num_layers": 2, | |
| "num_heads": 4, | |
| "dim_feedforward": 256, | |
| "max_length": 20, | |
| "dropout": 0.1, | |
| "id2label": { | |
| "0": "MALE", | |
| "1": "FEMALE" | |
| }, | |
| "label2id": { | |
| "MALE": 0, | |
| "FEMALE": 1 | |
| }, | |
| "metrics": { | |
| "best_val_accuracy": 0.8449828995420555, | |
| "best_val_f1": 0.8449560904215879 | |
| } | |
| } |