-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.json
58 lines (58 loc) · 1.14 KB
/
config.json
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
48
49
50
51
52
53
54
55
56
57
58
{
"DATA_DIR": "/mnt/fscompute_shared/Miller_AntarcticData/Specs_sequenced2",
"OUTPUT_DIR": "/mnt/fscompute_shared/Miller_AntarcticData/CNN+Noise_Results/",
"LOCATIONS": [
"BallenyIslands2015",
"casey2014",
"casey2017",
"ElephantIsland2013",
"ElephantIsland2014",
"kerguelen2005",
"kerguelen2014",
"kerguelen2015",
"Greenwich64S2015",
"MaudRise2014",
"RossSea2014"
],
"CATEGORIES": [
"20Plus",
"20Hz",
"A",
"B",
"D",
"Dswp",
"Z",
"Noise"
],
"CATEGORIES_TO_JOIN": {
"20Hz20Plus": [
"20Plus",
"20Hz"
],
"ABZ": [
"A",
"B",
"Z"
],
"DDswp": [
"D",
"Dswp"
]
},
"SAMPLES_PER_CLASS": 30000,
"NOISE_RATIO": [0.25],
"NOISE_RATIO_TEST": ["all"],
"USE_CORRECTED_DATASET": false,
"VALID_SPLIT": 0.3,
"TEST_SPLIT": "blocked",
"model_name": "IdilCNN",
"BATCH_SIZE": 32,
"EPOCHS": 100,
"early_stop": 20,
"monitoring_metric": "val_imbalanced_metric",
"monitoring_direction": "max",
"loss_function": "custom_cross_entropy",
"learning_rate": 0.001,
"CLASS_WEIGHTS": null,
"test_after_training": true
}