NER_span_test / config.json
AntoineBourgois's picture
Upload config.json
8bd82dc verified
Raw
History Blame Contribute Delete
1.59 kB
{
"NER_span_dataset_path": "coreference_datasets/DROC.json",
"encoder": "LSX-UniWue/ModernGBERT_1B",
"max_seq_len": 8192,
"overlapping_ratio": 0.5,
"max_span_width": 8,
"width_emb_dim": 64,
"mlp_hidden": 128,
"dropout": 0.3,
"focal_gamma": 2.0,
"max_epochs": 50,
"batch_size": 1,
"grad_accum": 1,
"lr_encoder": 5e-06,
"lr_head": 0.0002,
"warmup_ratio": 0.1,
"max_grad_norm": 1.0,
"weight_decay": 0.1,
"fp16": true,
"neg_ratio": 20,
"doc_eval_epochs": 1,
"doc_eval_batch": 32,
"allow_crossing": false,
"allow_nested": false,
"early_stopping_metric": "win_micro_f1",
"early_stopping_patience": 15,
"validation_strategy": "batch_split",
"validation_ratio": 0.15,
"evaluation_strategy": "test_split",
"best_model_selection_metric": "win_micro_f1",
"seed": 42,
"dataset_path": "lattice-nlp/DROC",
"dataset_name": "DROC",
"language": "de",
"output_dir": "DROC-ModernGBERT_1B-batch_split",
"label_to_id": {
"O": 0,
"PER": 1
},
"id_to_label": {
"0": "O",
"1": "PER"
},
"stride_tokens": 4096,
"window_tokens": 8192,
"trained_model_output_dir": "DROC-ModernGBERT_1B-batch_split/test",
"test_files": [
"Ebner-Eschenbach,-Marie-von__Agave",
"Ehrmann,-Marianne__Amalie. Eine wahre Geschichte in Briefen",
"Fontane,-Theodor__Der Stechlin",
"Fontane,-Theodor__Schach von Wuthenow",
"Ludwig,-Otto__Die Heiteretei und ihr Widerspiel",
"Mundt,-Theodor__Madonna. Unterhaltungen mit einer Heiligen",
"Otto,-Louise__Zweiter Band",
"Schopenhauer,-Johanna__Richard Wood"
]
}