| { | |
| "base_model": "bert-base-uncased", | |
| "architecture": "BERTGRUSentiment", | |
| "task": "sentiment-analysis", | |
| "label_mapping": { | |
| "0": "Negative", | |
| "1": "Positive" | |
| }, | |
| "checkpoint_files": { | |
| "inner": "transformer-model-inner.pt", | |
| "outer": "transformer-model-outer.pt" | |
| }, | |
| "model_hyperparameters": { | |
| "hidden_dim": 256, | |
| "output_dim": 1, | |
| "n_layers": 2, | |
| "bidirectional": true, | |
| "dropout": 0.2 | |
| }, | |
| "prediction": { | |
| "activation": "sigmoid", | |
| "threshold": 0.5 | |
| } | |
| } |