14 variables = [
'p',
'pz',
'daughter(0, p)',
'daughter(0, pz)',
'daughter(1, p)',
'daughter(1, pz)',
15 'chiProb',
'dr',
'dz',
'daughter(0, dr)',
'daughter(1, dr)',
'daughter(0, chiProb)',
'daughter(1, chiProb)',
16 'daughter(0, kaonID)',
'daughter(0, pionID)',
'daughterAngle(0, 1)']
18 if __name__ ==
"__main__":
21 if not (os.path.isfile(
'train.root')
and os.path.isfile(
'test.root')):
24 general_options = basf2_mva.GeneralOptions()
25 general_options.m_datafiles = basf2_mva.vector(
"train.root")
26 general_options.m_treename =
"tree"
27 general_options.m_variables = basf2_mva.vector(*variables)
28 general_options.m_target_variable =
"isSignal"
31 (
'Trivial.xml', basf2_mva.TrivialOptions()),
32 (
'Python.xml', basf2_mva.PythonOptions()),
33 (
'FastBDT.xml', basf2_mva.FastBDTOptions()),
34 (
'TMVAClassification.xml', basf2_mva.TMVAOptionsClassification()),
35 (
'FANN.xml', basf2_mva.FANNOptions()),
39 with tempfile.TemporaryDirectory()
as tempdir:
40 os.symlink(os.path.abspath(
'train.root'), tempdir +
'/' + os.path.basename(
'train.root'))
41 os.symlink(os.path.abspath(
'test.root'), tempdir +
'/' + os.path.basename(
'test.root'))
44 for identifier, specific_options
in methods:
45 general_options.m_identifier = identifier
46 basf2_mva.teacher(general_options, specific_options)
48 basf2_mva.expert(basf2_mva.vector(*[i
for i, _
in methods]),
49 basf2_mva.vector(
'train.root'),
'tree',
'expert.root')
51 subprocess.call(
'basf2_mva_evaluate.py -o latex.pdf -train train.root -data test.root -i ' +
52 ' '.join([i
for i, _
in methods]), shell=
True)
55 shutil.copyfile(tempdir +
'/latex.pdf',
'latex.pdf')