Belle II Software  release-08-01-10
092_cs.py
1 #!/usr/bin/env python3
2 
3 import basf2_mva
4 
5 general_options = basf2_mva.GeneralOptions()
6 general_options.m_datafiles = basf2_mva.vector("ContinuumSuppression.root")
7 general_options.m_treename = "tree"
8 general_options.m_identifier = "MVAFastBDT.root" # outputted weightfile
9 general_options.m_variables = basf2_mva.vector(
10  "R2",
11  "thrustBm",
12  "thrustOm",
13  "cosTBTO",
14  "cosTBz",
15  "KSFWVariables(et)",
16  "KSFWVariables(mm2)",
17  "KSFWVariables(hso00)",
18  "KSFWVariables(hso02)",
19  "KSFWVariables(hso04)",
20  "KSFWVariables(hso10)",
21  "KSFWVariables(hso12)",
22  "KSFWVariables(hso14)",
23  "KSFWVariables(hso20)",
24  "KSFWVariables(hso22)",
25  "KSFWVariables(hso24)",
26  "KSFWVariables(hoo0)",
27  "KSFWVariables(hoo1)",
28  "KSFWVariables(hoo2)",
29  "KSFWVariables(hoo3)",
30  "KSFWVariables(hoo4)",
31  "CleoConeCS(1)",
32  "CleoConeCS(2)",
33  "CleoConeCS(3)",
34  "CleoConeCS(4)",
35  "CleoConeCS(5)",
36  "CleoConeCS(6)",
37  "CleoConeCS(7)",
38  "CleoConeCS(8)",
39  "CleoConeCS(9)",
40 )
41 general_options.m_target_variable = "isContinuumEvent"
42 fastbdt_options = basf2_mva.FastBDTOptions()
43 
44 basf2_mva.teacher(general_options, fastbdt_options)