TridentGNNClassifier/diffpool_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
57
58
59
60
61

{
    "dataset" : {
        "type" : "h5",
        "paths" : {
            "train" : "/data/icecube/ssarkar/trident_gnn/dataset/resampled_dataset_4/dataset_split/train.h5",
            "validation" : "/data/icecube/ssarkar/trident_gnn/dataset/resampled_dataset_4/dataset_split/val.h5",
            "test" : "/data/icecube/ssarkar/trident_gnn/dataset/resampled_dataset_4/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" : "pyg_diffpool",
        "embed_dim" : 32,
	"pool_dim"  : 16,
        "hidden_embed_layers" : 1,
        "hidden_pool_layers"  : 1,
        "units_fully_connected" : [16,1],
        "num_pool_levels" : 2,
        "units_clusters" : [8,4],
        "use_batchnorm" : true,
        "use_residual" : false,
        "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
}