|
a |
|
b/default.json |
|
|
1 |
{ |
|
|
2 |
"data_dir" : "gs://storage.andrewrpatterson.com/datasets/debug", |
|
|
3 |
"output_dir" : "/Users/apatterson/Desktop/", |
|
|
4 |
"training_data" : "train.csv", |
|
|
5 |
"validation_data" :"validation.csv", |
|
|
6 |
"test_data" : "test.csv", |
|
|
7 |
"num_train_examples": 3835, |
|
|
8 |
"max_sequence_length": 512, |
|
|
9 |
"warmup_proportion": 0.1, |
|
|
10 |
"device": "cpu", |
|
|
11 |
"classifier" : "bert-general", |
|
|
12 |
"tokenizer" : "bert-base-uncased", |
|
|
13 |
"pretrained_model" : "bert-base-uncased", |
|
|
14 |
"experiment_name": "default", |
|
|
15 |
"epochs": 5, |
|
|
16 |
"train_batch_size": 16, |
|
|
17 |
"gradient_accumulation_steps": 4, |
|
|
18 |
"eval_batch_size": 16, |
|
|
19 |
"learning_rate": 0.01, |
|
|
20 |
"target": "readm_30d", |
|
|
21 |
"datareader": "one-doc", |
|
|
22 |
"vocab_size": 30000, |
|
|
23 |
"embed_size": 100, |
|
|
24 |
"num_layers": 12, |
|
|
25 |
"validation_metric": "roc", |
|
|
26 |
"evaluator": "std" |
|
|
27 |
} |