Transformer-Explainability/BERT_params/movies_baas.json

{
    "start_server": 0,
    "bert_dir": "model_components/uncased_L-12_H-768_A-12/",
    "max_length": 512,
    "pooling_strategy": "CLS_TOKEN",
    "evidence_identifier": {
        "batch_size": 64,
        "epochs": 3,
        "patience": 10,
        "lr": 1e-3,
        "max_grad_norm": 1.0,
        "sampling_method": "random",
        "sampling_ratio": 1.0
    },
    "evidence_classifier": {
        "classes": [ "NEG", "POS" ],
        "batch_size": 64,
        "epochs": 3,
        "patience": 10,
        "lr": 1e-3,
        "max_grad_norm": 1.0,
        "sampling_method": "everything"
    }
}