/
default_settings.json
59 lines (59 loc) · 1.84 KB
/
default_settings.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
59
{
"dataset" : {
"type" : "hd5",
"paths" : {
"train" : "../data/data_dragon6_shuffled/train.hd5",
"validation" : "../data/data_dragon6_shuffled/val.hd5",
"test" : "../data/data_dragon6_shuffled/test.hd5"
},
"features" : [
"FirstCharge", "CumulativeCharge", "Time",
"VertexX", "VertexY", "VertexZ",
"RecoX", "RecoY", "RecoZ",
"RecoAzimuth", "RecoZenith"
],
"graph_features" : null,
"coordinates" : [
"VertexX", "VertexY", "VertexZ"
],
"shuffle" : false,
"max_charge" : null,
"max_true_energy" : null,
"min_true_energy" : null,
"distances_precomputed" : false,
"balance_classes" : false,
"min_track_length" : null,
"max_cascade_energy" : null,
"class_weights" : "balanced",
"flavors" : null,
"currents" : null
},
"model" : {
"type" : "gcn",
"hidden_units_graph_convolutions" : [64, 64, 64],
"hidden_units_lstm" : [32, 16],
"hidden_units_fully_connected" : [32, 1],
"hidden_units_graph_mlp" : null,
"hidden_units_regression" : [64, 64, 5],
"use_batchnorm" : true,
"dropout_rate" : 0.5,
"use_residual" : true
},
"training" : {
"loss" : "binary_crossentropy",
"auxiliary_loss_lambda" : 1e-1,
"min_learning_rate" : 1e-6,
"learning_rate" : 5e-4,
"learning_rate_scheduler" : "reduce_on_plateau",
"learning_rate_scheduler_patience" : 2,
"learning_rate_decay" : 1e-5,
"batch_size" : 64,
"epochs" : 50,
"directory" : "./hd5_{0}/",
"checkpoint_period" : 1,
"use_class_weights" : true,
"logfile" : null,
"logdir" : "./log"
},
"seed" : null
}