ml-pid-cbm/ml_pid_cbm/config.json at main · julnow/ml-pid-cbm · 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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
{
  "file_names": {
    "training": "/Users/julnow/gsi/mgr/trees/PlainTree2M_DCM_tof_primaries_12agev.root",
    "test": "/Users/julnow/gsi/mgr/trees/PlainTree1M_URQMD_tof_primaries_12agev.root"
  },
  "var_names": {
    "momentum": "Complex_p",
    "charge": "Complex_q",
    "mass2": "Complex_mass2",
    "pid": "Complex_pid"
  },
  "features_for_train": [
    "Complex_v_tof",
    "Complex_t",
    "Complex_vtx_chi2",
    "Complex_chi2_ov_ndf_vtx"
  ],
  "vars_to_draw": [
    "Complex_E",
    "Complex_chi2_ov_ndf_vtx",
    "Complex_eta",
    "Complex_l",
    "Complex_t",
    "Complex_mass2",
    "Complex_p",
    "Complex_pT",
    "Complex_phi",
    "Complex_pT",
    "Complex_phi",
    "Complex_rapidity",
    "Complex_sim_p",
    "Complex_sim_pT",
    "Complex_sim_phi",
    "Complex_vtx_chi2",
    "Complex_M",
    "Complex_v_tof",
    "xgb_preds"
  ],
  "cuts": {
    "Complex_mass2": {
      "lower": -1.0,
      "upper": 2.0
    },
    "Complex_pT": {
      "lower": 0.0,
      "upper": 2.0
    },
    "Complex_p": {
      "lower": -12.0,
      "upper": 12.0
    },
    "Complex_eta": {
      "lower": 0.0,
      "upper": 6.0
    },
    "Complex_mother_id": {
      "lower": -2,
      "upper": 0
    },
    "Complex_vtx_chi2": {
      "lower": 0,
      "upper": 4000
    },
    "Complex_chi2_ov_ndf_vtx": {
      "lower": 0,
      "upper": 4000
    }
  },
  "hyper_params": {
    "values": {
      "n_estimators": 670,
      "max_depth": 5,
      "learning_rate": 0.04281706988524796,
      "gamma": 1,
      "subsample": 0.6296676786626392,
      "alpha": 0,
      "lambda": 1
    },
    "ranges": {
      "n_estimators": [
        300,
        1200
      ],
      "max_depth": [
        2,
        8
      ],
      "learning_rate": [
        0.01,
        0.1
      ],
      "gamma": [
        0,
        10
      ],
      "subsample": [
        0.3,
        0.9
      ],
      "alpha": [
        0,
        5
      ],
      "lambda": [
        0,
        10
      ]
    }
  }
}