9 parser = argparse.ArgumentParser(description=
'Dump a tree to text.')
10 parser.add_argument(
'input', type=str,
12 Input file. Will take "cvndump/CVNTrainTree" directory and dump it 15 args = parser.parse_args()
20 outfileName = infile.replace(
".root",
".cvnevents.txt")
22 outfile =
open(outfileName,
"w")
24 rootFile = TFile.Open(infile,
"READ")
27 tree = rootFile.Get(
"cvndump/CVNTrainTree")
29 numEntries = tree.GetEntries()
37 numInputs = entry.train.fPMap.fPE.size()
39 outfile.write(
" ".join([
str(numEntries),
str(numInputs),
str(numOutputs)]) +
"\n")
41 inputs = [
str(pe)
for pe
in entry.train.fPMap.fPE]
42 outfile.write(
" ".join(inputs) +
"\n")
43 outputs = [
"0"] * numOutputs
44 outputs[entry.train.fInt] =
str(1)
45 outfile.write(
" ".join(outputs) +
"\n")
procfile open("FD_BRL_v0.txt")