SUSY-Mass-Regression/config_set2.yaml at main · alex-yang-upenn/SUSY-Mass-Regression · 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
# Directory paths
RAW_DATA_DIRECTORY: raw_data_set2
PROCESSED_DATA_DIRECTORY: processed_data_set2
RAW_DATA_BACKGROUND_DIRECTORY: raw_data_background_set2
PROCESSED_DATA_BACKGROUND_DIRECTORY: processed_data_background_set2

# Evaluation data files
EVAL_DATA_FILES:
  - test_bA_bgX_bgqqqqY_bgqqqqqqqqlv_A2500_X1000_Y500.npz
  - test_bA_bgX_bgqqqqY_bgqqqqqqqqlv_A3300_X1400_Y550.npz
  - test_bA_bgX_bgqqqqY_bgqqqqqqqqlv_A4100_X1800_Y600.npz
  - test_bA_bgX_bgqqqqY_bgqqqqqqqqlv_A5300_X2400_Y800.npz
  - test_bA_bgX_bgqqqqY_bgqqqqqqqqlv_A6500_X3000_Y1000.npz

# Data structure parameters
DECAY_CHAIN: ["P1", "P2"]
N_PARTICLES: 12
N_FEATURES: 7
MET_IDS: [12]
LEPTON_IDS: [11]
GLUON_IDS: [21]
TRAIN_TEST_SPLIT: 0.2
SCALABLE_FEATURES: [0, 1, 2]

GNN_BASELINE_F_R_LAYER_SIZES: [96, 64, 32]
GNN_BASELINE_F_O_LAYER_SIZES: [128, 96, 64]
GNN_BASELINE_PHI_C_LAYER_SIZES: [32, 16, 8]
GNN_BASELINE_LEARNING_RATE: 0.0005
GNN_BASELINE_EARLY_STOPPING_PATIENCE: 6
GNN_BASELINE_REDUCE_LR_PATIENCE: 2

GNN_TRANSFORMED_LEARNING_RATE: 0.0005
GNN_TRANSFORMED_LEARNING_RATE_DECAY: 0.9

SIAMESE_PHI_C_LAYER_SIZES: [64]
SIAMESE_PROJ_HEAD_LAYER_SIZES: [96, 128]
SIAMESE_LEARNING_RATE: 0.0005

DOWNSTREAM_FREEZE_EPOCH: 15
DOWNSTREAM_LEARNING_RATE: 0.0005
DOWNSTREAM_FROZEN_LEARNING_RATE: 0.001
DOWNSTREAM_LR_DECAY: 0.90
DOWNSTREAM_NO_FINETUNE_LEARNING_RATE: 0.0001
DOWNSTREAM_LAYER_SIZES: [128, 64, 32, 16]

BATCHSIZE: 256
EPOCHS: 30
FINETUNE_EPOCHS: 25
SIAMESE_EPOCHS: 50
SIMCLR_LOSS_TEMP: 0.1

RUN_ID: 3
DATASET_NAME: "_set2"

ERR_BAR_H_SPACING: 20
BUCKET_WIDTH: 100

NUM_PARTICLES_TO_DELETE: 2