TridentGNNClassifier/default_settings.json at pyg_standalone · ssarkarbht/TridentGNNClassifier · GitHub

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

{
    "dataset" : {
        "type" : "h5",
        "paths" : {
            "train" : "/data/user/ssarkar/TridentProduction/reconstruction/trident_gnn/dataset/resampled_dataset_1/dataset_split/train.h5",
            "validation" : "/data/user/ssarkar/TridentProduction/reconstruction/trident_gnn/dataset/resampled_dataset_1/dataset_split/val.h5",
            "test" : "/data/user/ssarkar/TridentProduction/reconstruction/trident_gnn/dataset/resampled_dataset_1/dataset_split/test.h5"
        },
        "features" : [
            "ResidualTime", "PhotonTrackLength", "ChargeFirstPulse",
            "TrackDOMDistance", "TotalCharge", "TimeMaxPulse",
            "ChargeMaxPulse", "DeltaTimeStd"
        ],
        "graph_features": null,
        "edgeparams" : [
            "EdgePositionX", "EdgePositionY", "EdgePositionZ",
            "EdgeDirectionX", "EdgeDirectionY", "EdgeDirectionZ"
        ],
        "shuffle" : false,
        "max_charge" : null,
        "balance_classes" : false,
        "min_track_length" : null,
        "max_cascade_energy" : null,
        "distances_precomputed" : true,
	"flavors" : null,
	"currents" : null,
	"class_weights" : "balanced"
    },
    "model" : {
        "type" : "gcn",
        "hidden_units_graph_convolutions" : [16, 16, 16, 16],
        "hidden_units_fully_connected" : [1],
        "use_batchnorm" : true,
        "use_residual" : true,
        "dropout_rate" : 0.2
    },
    "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" : 32,
		    "epochs" : 50,
		    "directory" : "trained_model_database/hd5_{0}/",
		    "checkpoint_period" : 1,
		    "use_class_weights" : false,
		    "logfile" : null,
		    "logdir" : "./log"
		 },

    "seed" : null
}