File size: 1,214 Bytes
b1c5630
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
{
    "_CHeeSEArguments": "CHeeSEArguments",
    "model_name_or_path": "dbmdz/bert-base-german-cased",
    "dataset_path": "./dataset.py",
    "metric_path": "./metric.py",
    "task": "stance_detection",
    "first_sentence_inputs": [
        "question"
    ],
    "second_sentence_inputs": [
        "title",
        "snippet",
        "paragraphs"
    ],
    "labels_to_predict": [
        "stance"
    ],
    "_CHeeSETrainingArguments": "CHeeSETrainingArguments",
    "do_train": true,
    "do_eval": true,
    "do_predict": true,
    "do_cross_validation": false,
    "output_dir": "/netscratch/schnitzler/bert",
    "logging_dir": "/netscratch/schnitzler/bert",
    "save_total_limit": 2,
    "log_to_file": true,
    "logging_strategy": "steps",
    "logging_steps": 50,
    "cross_validation_folds": 5,
    "num_train_epochs": 4,
    "per_device_train_batch_size": 8,
    "per_device_eval_batch_size": 8,
    "fp16": false,
    "learning_rate": 3e-05,
    "prediction_csv_kwargs": {
        "index": false
    },
    "prediction_columns_to_include": [
        "title",
        "snippet",
        "paragraphs",
        "question",
        "labels",
        "stance"
    ],
    "overwrite_output_dir": true
}