IdeoBert-IT-classifier-v2 / training_args.json
paride92's picture
Upload folder using huggingface_hub
0a2fc22 verified
Raw
History Blame Contribute Delete
890 Bytes
{
"model_name": "EuroBERT/EuroBERT-610m",
"train_file": "train_set_binary.csv",
"test_file": "test_set_binary.csv",
"text_column": "sentence",
"target_column": "ideological",
"label2id": {
"0": 0,
"1": 1
},
"id2label": {
"0": "0",
"1": "1"
},
"seed": 42,
"max_seq_length": 128,
"num_train_epochs": 2,
"per_device_train_batch_size": 64,
"per_device_eval_batch_size": 64,
"gradient_accumulation_steps": 1,
"learning_rate": 1e-05,
"weight_decay": 0.02,
"warmup_ratio": 0.1,
"warmup_steps": 43,
"logging_steps": 50,
"metric_for_best_model": "f1",
"early_stopping_patience": 3,
"max_grad_norm": 1.0,
"use_class_weights": true,
"class_weights": [
1.0379174947738647,
0.9647552371025085
],
"classification_threshold": 0.5,
"gradient_checkpointing": true,
"dataloader_pin_memory": false,
"auto_empty_cache": true
}