confusionMatrix.C File Reference

#include <iostream.h>
#include <algorithm>
#include <math.h>

Go to the source code of this file.

Functions
int	maxID (double arrayInput[])
void	confusionMatrix (string input)

Function Documentation

void confusionMatrix

(

string

input

)

Definition at line 20 of file confusionMatrix.C.

References chisquared::c3, confusionMatrixTree::chosenlabel, om::cout, allTimeWatchdog::endl, MECModelEnuComparisons::i, calib::j, maxID(), nc, nevent, output, scale, plotROC::title, and confusionMatrixTree::truelabel.

                                   {
  gROOT->ForceStyle();
  
  TChain *ch=new TChain("t");
  ch->SetMakeClass(1); //<===== important                                                                                                       
  ch->Add(input.c_str());

  int truelabel=0;
  double numu=0;
  double nue=0;
  double nutau=0;
  double nc=0;
  double cosmic=0;
  int chosenlabel=0;

  double idOut[5]={0.,0.,0.,0.,0.};

  ch->SetBranchAddress("truelabel",&truelabel);
  ch->SetBranchAddress("numu",&numu);
  ch->SetBranchAddress("nue",&nue);
  ch->SetBranchAddress("nutau",&nutau);
  ch->SetBranchAddress("nc",&nc);
  ch->SetBranchAddress("cosmic",&cosmic);

  TH2D *heatmap= new TH2D( "heatmap", "", 5,0,5,5,0,5);  
  
  char *interaction[5] = 
    {"#nu_{#mu}", "#nu_{e}","#nu_{#tau}","NC","Cosmic"};
  
  //heatmap->SetBit(TH1::kCanRebin);
  Int_t nevent =ch.GetEntries();
  for(Int_t i=0;i<nevent;i++){
    ch->GetEntry(i);
    idOut[0]=numu;
    idOut[1]=nue;
    idOut[2]=nutau;
    idOut[3]=nc;
    idOut[4]=cosmic;
    chosenlabel=maxID(idOut);

    heatmap->Fill(double(truelabel),double(chosenlabel));

  }
  
  for (int i=1;i<6;i++){
    double scale=0;
    for (int j=1;j<6;j++){
      scale=scale+heatmap->GetBinContent(i,j);
    }

    std::cout<<"how many events do I see? "<<scale<<std::endl;
    
    for (int j=1;j<6;j++){
      if(heatmap->GetBinContent(i,j)/scale>0.0001)
        {
          heatmap->SetBinContent(i,j,(heatmap->GetBinContent(i,j))/scale);
        }
      else{
        heatmap->SetBinContent(i,j,.0001);
      }
    }   
  }

  

  string title="heatmap";
  
  TCanvas c3("c3",title.c_str(),900,900);
  
  heatmap->SetTitle("");
  heatmap->SetXTitle("True");
  heatmap->SetYTitle("Selected");
  for(int i=1;i<6;i++){
    heatmap->GetYaxis()->SetBinLabel(i, interaction[i-1]); 
    heatmap->GetXaxis()->SetBinLabel(i, interaction[i-1]); 
  }
  heatmap->GetXaxis()->SetTickLength(0);
  heatmap->GetYaxis()->SetTickLength(0);
  heatmap->GetYaxis()->SetTitleSize(0.06);
  heatmap->GetXaxis()->SetTitleSize(0.06);
  heatmap->GetYaxis()->SetTitleOffset(1);
  heatmap->GetXaxis()->SetTitleOffset(1);
  heatmap->GetYaxis()->SetLabelSize(0.025);
  heatmap->GetXaxis()->SetLabelSize(0.025);
  heatmap->GetZaxis()->SetLabelSize(0.03);

  heatmap->GetXaxis()->CenterTitle();
  heatmap->GetYaxis()->CenterTitle();
   
  gStyle->SetPalette(53);
  heatmap->SetMaximum(1.0);
  heatmap->Draw("colz");

  string output = "plots/"+title+".gif";
  c3.Print(output.c_str());
  string output = "plots/"+title+".C";
  c3.Print(output.c_str());
  string output = "plots/"+title+".eps";
  c3.Print(output.c_str());




  return;
}

int maxID

(

double

arrayInput[]

)

Definition at line 5 of file confusionMatrix.C.

References MECModelEnuComparisons::i, and cet::sqlite::max().

Referenced by cheat::BackTracker::ClusterToParticle(), confusionMatrix(), and cvn::HitObject().

                              {
  int max=0;
  double maxVal=0.;

  for(int i=0;i<5;i++){
    if(arrayInput[i]>maxVal){
      max=i;
      maxVal=arrayInput[i];
    }
    
  }
  
  return max;
}