Belle II Software  release-08-01-10
__init__.py
1 
10 PDG_TOKENS = [
11  0,
12  329,
13  110553,
14  5,
15  -12,
16  425,
17  553,
18  -20513,
19  -311,
20  13122,
21  323,
22  -11,
23  510,
24  -325,
25  20513,
26  -321,
27  110555,
28  100313,
29  10533,
30  310,
31  130553,
32  -10521,
33  20423,
34  521,
35  10521,
36  5114,
37  -13122,
38  -10533,
39  150,
40  -10511,
41  -13212,
42  5232,
43  -415,
44  -10213,
45  313,
46  -100313,
47  -23122,
48  20523,
49  15,
50  20333,
51  -100411,
52  -435,
53  2112,
54  -3314,
55  -413,
56  20213,
57  413,
58  82,
59  -2203,
60  -4,
61  9000553,
62  4422,
63  -541,
64  -3114,
65  -87,
66  30553,
67  100441,
68  -2214,
69  -10421,
70  -18,
71  13126,
72  3122,
73  2203,
74  9010443,
75  5322,
76  87,
77  3334,
78  10513,
79  -2103,
80  86,
81  -86,
82  -3212,
83  -20523,
84  14122,
85  -3126,
86  4432,
87  -2224,
88  515,
89  20313,
90  9000221,
91  -4424,
92  -10423,
93  -14122,
94  -10523,
95  -4412,
96  -431,
97  -4122,
98  4412,
99  10523,
100  -10411,
101  -4422,
102  -5114,
103  200555,
104  -30323,
105  10022,
106  -329,
107  130,
108  200551,
109  30323,
110  43122,
111  -43122,
112  10313,
113  -100413,
114  98,
115  -545,
116  10311,
117  -10431,
118  210553,
119  4122,
120  4132,
121  10431,
122  88,
123  -20323,
124  9910445,
125  321,
126  -4414,
127  3,
128  100413,
129  433,
130  -4334,
131  10411,
132  -2212,
133  -5103,
134  10553,
135  -13126,
136  -2,
137  -13,
138  5401,
139  1000020040,
140  3114,
141  30313,
142  -523,
143  83,
144  10113,
145  411,
146  85,
147  -5324,
148  220553,
149  100423,
150  -423,
151  53122,
152  113,
153  -513,
154  -211,
155  -20533,
156  10223,
157  480000000,
158  93,
159  20022,
160  -3214,
161  20113,
162  37,
163  3214,
164  4414,
165  335,
166  5132,
167  -319,
168  100213,
169  441,
170  10421,
171  5203,
172  -3203,
173  315,
174  -4332,
175  -20433,
176  -1000010020,
177  4103,
178  10543,
179  435,
180  5332,
181  -10541,
182  300553,
183  5224,
184  -2114,
185  3101,
186  -3101,
187  20433,
188  100553,
189  331,
190  -3216,
191  10423,
192  5214,
193  -323,
194  -14,
195  5222,
196  -315,
197  20443,
198  20413,
199  -5132,
200  -30363,
201  -411,
202  513,
203  44,
204  20533,
205  -3322,
206  3324,
207  -85,
208  97,
209  1,
210  9020443,
211  10511,
212  10333,
213  -4232,
214  -5,
215  4201,
216  -10323,
217  20223,
218  100223,
219  3224,
220  5201,
221  -20213,
222  30221,
223  -30313,
224  -30213,
225  325,
226  -10433,
227  10433,
228  20323,
229  -5332,
230  10213,
231  -213,
232  -543,
233  200553,
234  523,
235  13212,
236  -4201,
237  3203,
238  225,
239  -84,
240  -215,
241  84,
242  -4103,
243  7,
244  -5224,
245  -6,
246  6,
247  211,
248  -13124,
249  2114,
250  -521,
251  2103,
252  33,
253  -3303,
254  -4101,
255  -4301,
256  431,
257  -425,
258  17,
259  223,
260  11,
261  115,
262  8,
263  4301,
264  100211,
265  4203,
266  5212,
267  -3201,
268  -20423,
269  -4212,
270  120553,
271  5101,
272  4434,
273  -33122,
274  12,
275  94,
276  3201,
277  5334,
278  100551,
279  10443,
280  24,
281  215,
282  -10321,
283  557,
284  10555,
285  91,
286  423,
287  4114,
288  -327,
289  -82,
290  -4114,
291  -10313,
292  30223,
293  -8,
294  5122,
295  210551,
296  -3334,
297  5303,
298  1000010030,
299  33122,
300  -4403,
301  -13214,
302  531,
303  36,
304  -4124,
305  96,
306  100555,
307  95,
308  -20543,
309  92,
310  100221,
311  4312,
312  4403,
313  -4132,
314  -100421,
315  9000111,
316  2214,
317  3103,
318  -100211,
319  5503,
320  327,
321  -5303,
322  4303,
323  -5203,
324  100333,
325  4222,
326  30363,
327  -4434,
328  -3103,
329  34,
330  13214,
331  5103,
332  -511,
333  333,
334  511,
335  90,
336  4314,
337  -4324,
338  23122,
339  -3124,
340  -533,
341  25,
342  100421,
343  4424,
344  -41,
345  30443,
346  -525,
347  -3324,
348  213,
349  100323,
350  41,
351  30353,
352  -30353,
353  -3112,
354  4124,
355  4322,
356  -317,
357  3124,
358  -10531,
359  3216,
360  -15,
361  -3222,
362  319,
363  -20413,
364  525,
365  -1000020030,
366  -20313,
367  -9042413,
368  415,
369  23,
370  3322,
371  445,
372  10531,
373  -5334,
374  30343,
375  5403,
376  -5403,
377  4324,
378  -5222,
379  5301,
380  20553,
381  4332,
382  110551,
383  -421,
384  -30343,
385  -4203,
386  311,
387  100443,
388  -515,
389  13,
390  -5312,
391  -1000010030,
392  2,
393  5324,
394  2101,
395  -5401,
396  317,
397  -2101,
398  30213,
399  -4432,
400  3222,
401  -4112,
402  4212,
403  1000020030,
404  -3122,
405  421,
406  20555,
407  9000443,
408  350,
409  5312,
410  -4322,
411  -5201,
412  -44,
413  4112,
414  -100323,
415  99,
416  35,
417  -5301,
418  3112,
419  -5214,
420  -10513,
421  -3,
422  -5322,
423  10551,
424  10413,
425  443,
426  221,
427  81,
428  -24,
429  14,
430  2212,
431  -3224,
432  4334,
433  5314,
434  111,
435  -433,
436  9000211,
437  18,
438  4232,
439  10321,
440  -100213,
441  551,
442  3303,
443  -1,
444  -5314,
445  -313,
446  -531,
447  1114,
448  -2112,
449  -34,
450  10541,
451  535,
452  9920443,
453  -4222,
454  -535,
455  2224,
456  120555,
457  10441,
458  43,
459  -4312,
460  555,
461  20543,
462  1000010020,
463  -53122,
464  543,
465  545,
466  3314,
467  9030221,
468  -1114,
469  3212,
470  9042413,
471  541,
472  533,
473  100113,
474  -1000020040,
475  1103,
476  3312,
477  3126,
478  -5212,
479  -5112,
480  4101,
481  -4303,
482  10323,
483  -37,
484  -10311,
485  -4314,
486  -3312,
487  -100423,
488  -1103,
489  -7,
490  -9000211,
491  -16,
492  32,
493  21,
494  16,
495  4,
496  100557,
497  13124,
498  9010221,
499  100111,
500  4214,
501  23212,
502  -10413,
503  -5101,
504  -10543,
505  100411,
506  -4214,
507  -5122,
508  30113,
509  -5503,
510  5112,
511  -23212,
512  -4224,
513  4224,
514  530,
515  -17,
516  -5232,
517  22,
518  30643,
519  -30643,
520  30653,
521  -30653,
522  9000113,
523  9000213,
524  9020221,
525  10111,
526  10211,
527  100331,
528  9010113,
529  9010213,
530  10225,
531  227,
532  10115,
533  10215,
534  117,
535  217,
536  10331,
537  9010111,
538  9010211,
539  337,
540  9050225,
541  9060225,
542  119,
543  219,
544  229,
545  9080225,
546  9090225,
547  10315,
548  10325,
549  20315,
550  20325,
551 ]
552 
553 TOKENIZE_DICT = {v: k for k, v in enumerate(PDG_TOKENS)}
554 
555 # preprocessing configurations
556 PREPROC_CONFIG = {
557  "cuts": ["abs(x) < 10", "abs(y) < 10", "abs(z) < 10"],
558  "features": ['PDG', 'prodTime', "energy", 'x', 'y', 'z', 'px', 'py', 'pz']
559 }
560 
561 # model configurations
562 MODEL_CONFIG = {
563  "units": 32, "attention_heads": 4, "n_layers": 6, "use_gap": True
564 }