void trainBDTG_theta3(void) { TMVA::Tools::Instance(); auto outputFile = TFile::Open("TMVAOutputBDT_theta3.root", "RECREATE"); TMVA::Factory factory("TMVARegression", outputFile, "!V:!Silent:Color:DrawProgressBar:AnalysisType=Regression"); TMVA::DataLoader loader("dataset"); loader.AddVariable("L01x"); loader.AddVariable("L02x"); loader.AddVariable("L03x"); loader.AddVariable("L04x"); loader.AddVariable("L05x"); loader.AddVariable("L06x"); loader.AddVariable("L07x"); loader.AddVariable("L08x"); loader.AddVariable("L09x"); loader.AddVariable("L10x"); loader.AddVariable("L11x"); loader.AddVariable("L12x"); loader.AddVariable("L13x"); loader.AddVariable("L14x"); loader.AddVariable("L15x"); loader.AddVariable("L16x"); loader.AddVariable("L17x"); loader.AddVariable("L18x"); loader.AddVariable("L19x"); loader.AddVariable("L20x"); loader.AddVariable("L21x"); loader.AddVariable("L22x"); loader.AddVariable("L23x"); loader.AddVariable("L24x"); loader.AddVariable("L01y"); loader.AddVariable("L02y"); loader.AddVariable("L03y"); loader.AddVariable("L04y"); loader.AddVariable("L05y"); loader.AddVariable("L06y"); loader.AddVariable("L07y"); loader.AddVariable("L08y"); loader.AddVariable("L09y"); loader.AddVariable("L10y"); loader.AddVariable("L11y"); loader.AddVariable("L12y"); loader.AddVariable("L13y"); loader.AddVariable("L14y"); loader.AddVariable("L15y"); loader.AddVariable("L16y"); loader.AddVariable("L17y"); loader.AddVariable("L18y"); loader.AddVariable("L19y"); loader.AddVariable("L20y"); loader.AddVariable("L21y"); loader.AddVariable("L22y"); loader.AddVariable("L23y"); loader.AddVariable("L24y"); loader.AddTarget("target := theta"); auto chain = new TChain("t"); chain->Add("../DATA/hd_root_*.root"); TCut mycuts("p>0.200 && p<5.0 && theta>0.017 && theta<0.2"); loader.AddRegressionTree( chain, 1.0 ); loader.PrepareTrainingAndTestTree( mycuts, "nTrain_Regression=1000000:nTest_Regression=500000:SplitMode=Random:NormMode=NumEvents:!V"); factory.BookMethod( &loader, TMVA::Types::kBDT, "BDTG", TString("!H:!V:Ntrees=4096::BoostType=Grad:Shrinkage=0.1:ncuts=20:MaxDepth=32:")+ TString("RegressionLossFunctionBDTG=AbsoluteDeviation")); factory.TrainAllMethods(); factory.TestAllMethods(); factory.EvaluateAllMethods(); outputFile->Close(); }