本文整理汇总了Python中trainer.Trainer.setSTreeName方法的典型用法代码示例。如果您正苦于以下问题：Python Trainer.setSTreeName方法的具体用法？Python Trainer.setSTreeName怎么用？Python Trainer.setSTreeName使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类trainer.Trainer的用法示例。
在下文中一共展示了Trainer.setSTreeName方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: selection
# 需要导入模块: from trainer import Trainer [as 别名]
# 或者: from trainer.Trainer import setSTreeName [as 别名]
#samples have a name, a color, a path, and a selection (not implemented yet for training)
#only the path is really relevant atm
cat='6j4t'
#change path if necessary
signal_test=Sample('t#bar{t}H test',ROOT.kBlue,"../../atlas-higgs-challenge-2014-v2.root",'')
signal_train=Sample('t#bar{t}H training',ROOT.kGreen,"../../atlas-higgs-challenge-2014-v2.root",'')
background_test=Sample('t#bar{t} test',ROOT.kRed+1,"../../atlas-higgs-challenge-2014-v2.root",'')
background_train=Sample('t#bar{t} training',ROOT.kRed-1,"../../atlas-higgs-challenge-2014-v2.root",'')
#create trainer and set trainer options
trainer=Trainer(variables,addtional_variables)
#trainer.addSamples(signal_train,background_train)
trainer.addSamples(signal_train,background_train,signal_test,background_test) #add the sample defined above
trainer.setSTreeName('signal') # name of signaltree in files
trainer.setBTreeName('background') # name of backgroundtree in files
trainer.setReasonableDefaults() # set some configurations to reasonable values
trainer.setEqualNumEvents(True) # reweight events so that integral in training and testsample is the same
trainer.useTransformations(True) # faster this way
trainer.setVerbose(True) # no output during BDT training and testing
trainer.setWeightExpression('Weight')
#set BDT options
trainer.setBDTOption("NTrees=10")
trainer.setBDTOption("Shrinkage=0.01")
trainer.setBDTOption("nCuts=50")
trainer.setBDTOption("MaxDepth=2")
print trainer.best_variables
trainer.trainBDT(variables)