261 UInt_t nevts = events.size();
264 Log() << kFATAL <<
"No events available to find min, max, mean and rms" <<
Endl;
267 const UInt_t nvar = events[0]->GetNVariables();
268 const UInt_t ntgt = events[0]->GetNTargets();
276 for (
Int_t cls=0; cls<fNumC; cls++) {
280 varMin[cls] =
new Double_t[nvar+ntgt];
281 varMax[cls] =
new Double_t[nvar+ntgt];
282 for (
UInt_t ivar=0; ivar<nvar+ntgt; ivar++) {
283 x0[cls][ivar] =
x2[cls][ivar] = 0;
284 varMin[cls][ivar] = DBL_MAX;
285 varMax[cls][ivar] = -DBL_MAX;
289 for (
UInt_t ievt=0; ievt<nevts; ievt++) {
290 const Event* ev = events[ievt];
294 sumOfWeights[cls] += weight;
295 if (fNumC > 1 ) sumOfWeights[fNumC-1] += weight;
296 for (
UInt_t var_tgt = 0; var_tgt < 2; var_tgt++ ){
297 UInt_t nloop = ( var_tgt==0?nvar:ntgt );
298 for (
UInt_t ivar=0; ivar<nloop; ivar++) {
301 if (
x < varMin[cls][(var_tgt*nvar)+ivar]) varMin[cls][(var_tgt*nvar)+ivar]=
x;
302 if (
x > varMax[cls][(var_tgt*nvar)+ivar]) varMax[cls][(var_tgt*nvar)+ivar]=
x;
304 x0[cls][(var_tgt*nvar)+ivar] +=
x*weight;
305 x2[cls][(var_tgt*nvar)+ivar] +=
x*
x*weight;
308 if (
x < varMin[fNumC-1][(var_tgt*nvar)+ivar]) varMin[fNumC-1][(var_tgt*nvar)+ivar]=
x;
309 if (
x > varMax[fNumC-1][(var_tgt*nvar)+ivar]) varMax[fNumC-1][(var_tgt*nvar)+ivar]=
x;
311 x0[fNumC-1][(var_tgt*nvar)+ivar] +=
x*weight;
312 x2[fNumC-1][(var_tgt*nvar)+ivar] +=
x*
x*weight;
320 for (
UInt_t var_tgt = 0; var_tgt < 2; var_tgt++ ){
321 UInt_t nloop = ( var_tgt==0?nvar:ntgt );
322 for (
UInt_t ivar=0; ivar<nloop; ivar++) {
323 for (
Int_t cls = 0; cls < fNumC; cls++) {
324 Double_t mean = x0[cls][(var_tgt*nvar)+ivar]/sumOfWeights[cls];
326 AddStats(cls, (var_tgt*nvar)+ivar, mean, rms, varMin[cls][(var_tgt*nvar)+ivar], varMax[cls][(var_tgt*nvar)+ivar]);
333 UInt_t maxL = 8, maxV = 0;
334 std::vector<UInt_t> vLengths;
335 for (
UInt_t ivar=0; ivar<nvar+ntgt; ivar++) {
343 UInt_t clen = maxL + 4*maxV + 11;
349 Log() << std::setw(maxL) <<
"Variable";
350 Log() <<
" " << std::setw(maxV) <<
"Mean";
351 Log() <<
" " << std::setw(maxV) <<
"RMS";
352 Log() <<
" " << std::setw(maxV) <<
"[ Min ";
353 Log() <<
" " << std::setw(maxV) <<
" Max ]"<<
Endl;
354 for (
UInt_t i=0; i<clen; i++) Log() <<
"-";
359 for (
UInt_t ivar=0; ivar<nvar+ntgt; ivar++) {
361 Log() << std::setw(maxL) << Variable(ivar).GetLabel() <<
":";
363 Log() << std::setw(maxL) << Target(ivar-nvar).GetLabel() <<
":";
364 Log() << std::setw(maxV) <<
Form(
format.Data(), GetMean(ivar) );
365 Log() << std::setw(maxV) <<
Form(
format.Data(), GetRMS(ivar) );
366 Log() <<
" [" << std::setw(maxV) <<
Form(
format.Data(), GetMin(ivar) );
367 Log() << std::setw(maxV) <<
Form(
format.Data(), GetMax(ivar) ) <<
" ]";
370 for (
UInt_t i=0; i<clen; i++) Log() <<
"-";
374 delete[] sumOfWeights;
375 for (
Int_t cls=0; cls<fNumC; cls++) {
378 delete [] varMin[cls];
379 delete [] varMax[cls];
463 if (fRootBaseDir==0 && theDirectory == 0)
return;
465 Log() << kDEBUG <<
"Plot event variables for ";
467 else Log() << GetName() <<
Endl;
471 if (theDirectory == 0) {
473 transfType += GetName();
478 const UInt_t nvar = fDataSetInfo.GetNVariables();
479 const UInt_t ntgt = fDataSetInfo.GetNTargets();
480 const Int_t ncls = fDataSetInfo.GetNClasses();
484 std::vector<std::vector<TH1*> > hVars( ncls );
485 std::vector<std::vector<std::vector<TH2F*> > > mycorr( ncls );
486 std::vector<std::vector<std::vector<TProfile*> > > myprof( ncls );
488 for (
Int_t cls = 0; cls < ncls; cls++) {
489 hVars.at(cls).resize ( nvar+ntgt );
490 hVars.at(cls).assign ( nvar+ntgt, 0 );
491 mycorr.at(cls).resize( nvar+ntgt );
492 myprof.at(cls).resize( nvar+ntgt );
493 for (
UInt_t ivar=0; ivar < nvar+ntgt; ivar++) {
494 mycorr.at(cls).at(ivar).resize( nvar+ntgt );
495 myprof.at(cls).at(ivar).resize( nvar+ntgt );
496 mycorr.at(cls).at(ivar).assign( nvar+ntgt, 0 );
497 myprof.at(cls).at(ivar).assign( nvar+ntgt, 0 );
504 if (nvar+ntgt > (
UInt_t)
gConfig().GetVariablePlotting().fMaxNumOfAllowedVariablesForScatterPlots) {
505 Int_t nhists = (nvar+ntgt)*(nvar+ntgt - 1)/2;
507 Log() << kINFO <<
"<PlotVariables> Will not produce scatter plots ==> " <<
Endl;
509 <<
"| The number of " << nvar <<
" input variables and " << ntgt <<
" target values would require "
510 << nhists <<
" two-dimensional" <<
Endl;
512 <<
"| histograms, which would occupy the computer's memory. Note that this" <<
Endl;
514 <<
"| suppression does not have any consequences for your analysis, other" <<
Endl;
516 <<
"| than not disposing of these scatter plots. You can modify the maximum" <<
Endl;
518 <<
"| number of input variables allowed to generate scatter plots in your" <<
Endl;
519 Log() <<
"| script via the command line:" <<
Endl;
521 <<
"| \"(TMVA::gConfig().GetVariablePlotting()).fMaxNumOfAllowedVariablesForScatterPlots = <some int>;\""
524 Log() << kINFO <<
"Some more output" <<
Endl;
531 for (
UInt_t var_tgt = 0; var_tgt < 2; var_tgt++) {
532 UInt_t nloops = ( var_tgt == 0? nvar:ntgt );
533 for (
UInt_t ivar=0; ivar<nloops; ivar++) {
534 const VariableInfo& info = ( var_tgt == 0 ? Variable( ivar ) : Target(ivar) );
537 Double_t mean = fVariableStats.at(fNumC-1).at( ( var_tgt*nvar )+ivar).fMean;
538 Double_t rms = fVariableStats.at(fNumC-1).at( ( var_tgt*nvar )+ivar).fRMS;
540 for (
Int_t cls = 0; cls < ncls; cls++) {
542 TString className = fDataSetInfo.GetClassInfo(cls)->GetName();
545 className += (ntgt == 1 && var_tgt == 1 ?
"_target" :
"");
573 h->GetXaxis()->SetTitle(
gTools().GetXTitleWithUnit( GetVariableAxisTitle( info ), info.
GetUnit() ) );
575 hVars.at(cls).at((var_tgt*nvar)+ivar) =
h;
578 if (nvar+ntgt <= (
UInt_t)
gConfig().GetVariablePlotting().fMaxNumOfAllowedVariablesForScatterPlots) {
580 for (
UInt_t v_t = 0; v_t < 2; v_t++) {
581 UInt_t nl = ( v_t==0?nvar:ntgt );
582 UInt_t start = ( v_t==0? (var_tgt==0?ivar+1:0):(var_tgt==0?nl:ivar+1) );
583 for (
UInt_t j=start; j<nl; j++) {
585 const VariableInfo& infoj = ( v_t == 0 ? Variable( j ) : Target(j) );
588 Double_t rxmin = fVariableStats.at(fNumC-1).at( ( v_t*nvar )+ivar).fMin;
589 Double_t rxmax = fVariableStats.at(fNumC-1).at( ( v_t*nvar )+ivar).fMax;
590 Double_t rymin = fVariableStats.at(fNumC-1).at( ( v_t*nvar )+j).fMin;
591 Double_t rymax = fVariableStats.at(fNumC-1).at( ( v_t*nvar )+j).fMax;
595 className.
Data(), transfType.
Data() ),
597 className.
Data(), transfType.
Data() ),
598 nbins2D, rxmin , rxmax,
599 nbins2D, rymin , rymax );
603 mycorr.at(cls).at((var_tgt*nvar)+ivar).at((v_t*nvar)+j) = h2;
611 className.
Data(), transfType.
Data() ), nbins1D,
615 p->GetXaxis()->SetTitle(
gTools().GetXTitleWithUnit( GetVariableAxisTitle( info ), info .GetUnit() ) );
616 p->GetYaxis()->SetTitle(
gTools().GetXTitleWithUnit( GetVariableAxisTitle( infoj ), infoj.
GetUnit() ) );
617 myprof.at(cls).at((var_tgt*nvar)+ivar).at((v_t*nvar)+j) =
p;
625 UInt_t nevts = events.size();
628 std::vector<Double_t> xregmean ( nvar+1, 0 );
629 std::vector<Double_t> x2regmean( nvar+1, 0 );
630 std::vector<Double_t> xCregmean( nvar+1, 0 );
633 for (
UInt_t ievt=0; ievt<nevts; ievt++) {
635 const Event* ev = events[ievt];
643 xregmean[nvar] += valr;
644 x2regmean[nvar] += valr*valr;
645 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
647 xregmean[ivar] += vali;
648 x2regmean[ivar] += vali*vali;
649 xCregmean[ivar] += vali*valr;
654 for (
UInt_t var_tgt = 0; var_tgt < 2; var_tgt++) {
655 UInt_t nloops = ( var_tgt == 0? nvar:ntgt );
656 for (
UInt_t ivar=0; ivar<nloops; ivar++) {
660 hVars.at(cls).at( ( var_tgt*nvar )+ivar)->Fill( vali, weight );
665 for (
UInt_t v_t = 0; v_t < 2; v_t++) {
666 UInt_t nl = ( v_t==0 ? nvar : ntgt );
667 UInt_t start = ( v_t==0 ? (var_tgt==0?ivar+1:0) : (var_tgt==0?nl:ivar+1) );
668 for (
UInt_t j=start; j<nl; j++) {
670 mycorr.at(cls).at( ( var_tgt*nvar )+ivar).at( ( v_t*nvar )+j)->Fill( vali, valj, weight );
671 myprof.at(cls).at( ( var_tgt*nvar )+ivar).at( ( v_t*nvar )+j)->Fill( vali, valj, weight );
681 for (
UInt_t ivar=0; ivar<=nvar; ivar++) {
682 xregmean[ivar] /= nevts;
683 x2regmean[ivar] = x2regmean[ivar]/nevts - xregmean[ivar]*xregmean[ivar];
685 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
686 xCregmean[ivar] = xCregmean[ivar]/nevts - xregmean[ivar]*xregmean[nvar];
687 xCregmean[ivar] /=
TMath::Sqrt( x2regmean[ivar]*x2regmean[nvar] );
690 fRanking.push_back(
new Ranking( GetName() +
"Transformation",
"|Correlation with target|" ) );
691 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
693 fRanking.back()->AddRank(
Rank( fDataSetInfo.GetVariableInfo(ivar).GetLabel(), abscor ) );
696 if (nvar+ntgt <= (
UInt_t)
gConfig().GetVariablePlotting().fMaxNumOfAllowedVariablesForScatterPlots) {
699 fRanking.push_back(
new Ranking( GetName() +
"Transformation",
"Mutual information" ) );
700 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
701 TH2F*
h1 = mycorr.at(0).at( nvar ).at( ivar );
703 fRanking.back()->AddRank(
Rank( fDataSetInfo.GetVariableInfo(ivar).GetLabel(), mi ) );
707 fRanking.push_back(
new Ranking( GetName() +
"Transformation",
"Correlation Ratio" ) );
708 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
709 TH2F* h2 = mycorr.at(0).at( nvar ).at( ivar );
711 fRanking.back()->AddRank(
Rank( fDataSetInfo.GetVariableInfo(ivar).GetLabel(), cr ) );
715 fRanking.push_back(
new Ranking( GetName() +
"Transformation",
"Correlation Ratio (T)" ) );
716 for (
UInt_t ivar=0; ivar<nvar; ivar++) {
719 fRanking.back()->AddRank(
Rank( fDataSetInfo.GetVariableInfo(ivar).GetLabel(), cr ) );
726 else if (fDataSetInfo.GetNClasses() == 2
727 && fDataSetInfo.GetClassInfo(
"Signal") != NULL
728 && fDataSetInfo.GetClassInfo(
"Background") != NULL
730 fRanking.push_back(
new Ranking( GetName() +
"Transformation",
"Separation" ) );
731 for (
UInt_t i=0; i<nvar; i++) {
733 hVars.at(fDataSetInfo.GetClassInfo(
"Background")->GetNumber()).at(i) );
734 fRanking.back()->AddRank(
Rank( hVars.at(fDataSetInfo.GetClassInfo(
"Signal")->GetNumber()).at(i)->GetTitle(),
744 if (theDirectory == 0) {
750 outputDir +=
"_" +
TString(trf->GetShortName());
752 TString uniqueOutputDir = outputDir;
755 while( (o = fRootBaseDir->
FindObject(uniqueOutputDir)) !=
nullptr ){
757 Log() << kINFO <<
"A " << o->
ClassName() <<
" with name " << o->
GetName() <<
" already exists in "
758 << fRootBaseDir->GetPath() <<
", I will try with "<<uniqueOutputDir<<
"." <<
Endl;
767 localDir = fRootBaseDir->
mkdir( uniqueOutputDir );
770 Log() << kVERBOSE <<
"Create and switch to directory " << localDir->
GetPath() <<
Endl;
776 for (
UInt_t i=0; i<nvar+ntgt; i++) {
777 for (
Int_t cls = 0; cls < ncls; cls++) {
778 if (hVars.at(cls).at(i) != 0) {
779 hVars.at(cls).at(i)->Write();
780 hVars.at(cls).at(i)->SetDirectory(
nullptr);
781 delete hVars.at(cls).at(i);
787 if (nvar+ntgt <= (
UInt_t)
gConfig().GetVariablePlotting().fMaxNumOfAllowedVariablesForScatterPlots) {
789 localDir = localDir->
mkdir(
"CorrelationPlots" );
791 Log() << kDEBUG <<
"Create scatter and profile plots in target-file directory: " <<
Endl;
795 for (
UInt_t i=0; i<nvar+ntgt; i++) {
796 for (
UInt_t j=i+1; j<nvar+ntgt; j++) {
797 for (
Int_t cls = 0; cls < ncls; cls++) {
798 if (mycorr.at(cls).at(i).at(j) != 0 ) {
799 mycorr.at(cls).at(i).at(j)->Write();
800 mycorr.at(cls).at(i).at(j)->SetDirectory(
nullptr);
801 delete mycorr.at(cls).at(i).at(j);
803 if (myprof.at(cls).at(i).at(j) != 0) {
804 myprof.at(cls).at(i).at(j)->Write();
805 myprof.at(cls).at(i).at(j)->SetDirectory(
nullptr);
806 delete myprof.at(cls).at(i).at(j);
812 if (theDirectory != 0 ) theDirectory->
cd();
813 else fRootBaseDir->cd();