TJetNet Class Reference

#include <TJetNet.h>

Inheritance diagram for TJetNet:

Collaboration diagram for TJetNet:

Public Types
enum	TActivationFunction {   afSigmoid = 1 , afTanh = 2 , afExp = 3 , afLinear = 4 ,   afSigmoidEntropy = 5 }

Public Member Functions
	TJetNet (void)
	TJetNet (Int_t aTestCount, Int_t aTrainCount, const Int_t aLayersCnt, const Int_t *aLayers)
virtual	~TJetNet (void)
void	Print (void)
Int_t	GetTrainSetCnt (void) const
Int_t	GetTestSetCnt (void) const
Int_t	GetInputDim (void) const
Int_t	GetHiddenLayerDim (void) const
Int_t	GetHiddenLayerSize (Int_t number) const
Int_t	GetOutputDim (void) const
void	SetInputTrainSet (Int_t aPatternInd, Int_t aInputInd, Double_t aValue)
void	SetOutputTrainSet (Int_t aPatternInd, Int_t aOutputInd, Double_t aValue)
void	SetInputTestSet (Int_t aPatternInd, Int_t aInputInd, Double_t aValue)
void	SetOutputTestSet (Int_t aPatternInd, Int_t aOutputInd, Double_t aValue)
void	SetEventWeightTrainSet (Int_t aPatternInd, Double_t aValue)
void	SetEventWeightTestSet (Int_t aPatternInd, Double_t aValue)
Double_t	GetInputTrainSet (Int_t aPatternInd, Int_t aInputInd)
Double_t	GetOutputTrainSet (Int_t aPatternInd, Int_t aOutputInd)
Double_t	GetInputTestSet (Int_t aPatternInd, Int_t aInputInd)
Double_t	GetOutputTestSet (Int_t aPatternInd, Int_t aOutputInd)
Double_t	GetEventWeightTrainSet (Int_t aPatternInd)
Double_t	GetEventWeightTestSet (Int_t aPatternInd)
Double_t	GetWeight (Int_t aLayerInd, Int_t aNodeInd, Int_t aConnectedNodeInd) const
Double_t	GetThreshold (Int_t aLayerInd, Int_t aNodeInd) const
Int_t	GetEpochs (void)
void	SetEpochs (const Int_t aEpochs)
void	Init (void)
Double_t	Train (void)
Int_t	Epoch (void)
Double_t	Test (void)
Double_t	TestBTAG (void)
void	Shuffle (Bool_t aShuffleTrainSet=true, Bool_t aShuffleTestSet=true)
void	SaveDataAscii (TString aFileName="jndata.dat")
void	SaveDataRoot (TString aFileName="jndata.root")
void	LoadDataAscii (TString aFileName="jndata.dat")
void	LoadDataRoot (TString aFileName="jndata.root")
void	DumpToFile (TString aFileName="fort.8")
void	ReadFromFile (TString aFileName="fort.8")
Double_t	GetOutput (Int_t aIndex=0)
void	SetInputs (Int_t aIndex=0, Double_t aValue=0.0)
void	Evaluate (Int_t aPattern)
void	Evaluate ()
void	writeNetworkInfo (Int_t typeOfInfo=0)
Int_t	GetLayerCount (void)
Int_t	GetUnitCount (Int_t aLayer)
void	SelectiveFields (Int_t aLayerA, Int_t aNodeA1, Int_t aNodeA2, Int_t aNodeB1, Int_t aNodeB2, Int_t aSwitch=0)
void	SetUpdatesPerEpoch (Int_t aValue)
void	SetUpdatingProcedure (Int_t aValue)
void	SetErrorMeasure (Int_t aValue)
void	SetActivationFunction (Int_t aValue)
void	SetActivationFunction (Int_t aValue, Int_t layerN)
void	SetPatternsPerUpdate (Int_t aValue)
void	SetLearningRate (Double_t aValue)
void	SetMomentum (Double_t aValue)
void	SetInitialWeightsWidth (Double_t aValue)
void	SetLearningRateDecrease (Double_t aValue)
void	SetPottsUnits (Int_t aValue)
Int_t	GetUpdatesPerEpoch (void)
Int_t	GetUpdatingProcedure (void)
Int_t	GetErrorMeasure (void)
Int_t	GetActivationFunction (void) const
Int_t	GetActivationFunction (Int_t layerN) const
Int_t	GetPatternsPerUpdate (void)
Double_t	GetLearningRate (void)
Double_t	GetMomentum (void)
Double_t	GetInitialWeightsWidth (void)
Double_t	GetLearningRateDecrease (void)
Int_t	GetPottsUnits ()
void	LockInit (void)
void	UnlockInit (void)
Int_t	GetMSTJN (Int_t aIndex)
Double_t	GetPARJN (Int_t aIndex)
void	SetMSTJN (Int_t aIndex, Int_t aValue)
void	SetPARJN (Int_t aIndex, Double_t aValue)
void	Normalize (void)
void	Randomize (void)
TTrainedNetwork *	createTrainedNetwork () const
void	readBackTrainedNetwork (const TTrainedNetwork *)
void	NormalizeOutput (bool isTrue=true)

Private Member Functions
void	mSetWeight (Double_t weight, Int_t aLayerInd, Int_t aNodeInd, Int_t aConnectedNodeInd)
void	mSetThreshold (Double_t threshold, Int_t aLayerInd, Int_t aNodeInd)
Int_t	CopyFile (TString aSrcFile, TString aDestFile)
void	Reinitialize (void)

Private Attributes
TActivationFunction	m_enActFunction {}
Int_t	m_LayerCount
Int_t *	m_pLayers
TNeuralDataSet *	m_pInputTrainSet
	Array which contains the number of units in each layer.
TNeuralDataSet *	m_pOutputTrainSet
TNeuralDataSet *	m_pInputTestSet
TNeuralDataSet *	m_pOutputTestSet
Int_t	m_TrainSetCnt
Int_t	m_TestSetCnt
Int_t	m_InputDim
Int_t	m_HiddenLayerDim
Int_t	m_OutputDim
Int_t	m_Epochs
Int_t	m_CurrentEpoch
Bool_t	m_Debug
Bool_t	m_IsInitialized
Bool_t	m_InitLocked
Bool_t	m_NormalizeOutput

Detailed Description

Definition at line 40 of file TJetNet.h.

Member Enumeration Documentation

TActivationFunction

enum TJetNet::TActivationFunction

Enumerator
afSigmoid
afTanh
afExp
afLinear
afSigmoidEntropy

Definition at line 150 of file TJetNet.h.

                             {
      afSigmoid = 1,
      afTanh    = 2,
      afExp     = 3,
      afLinear  = 4,
      afSigmoidEntropy = 5
    };

Constructor & Destructor Documentation

TJetNet() [1/2]

TJetNet::TJetNet

(

void

)

TJetNet() [2/2]

TJetNet::TJetNet	(	Int_t	aTestCount,
		Int_t	aTrainCount,
		const Int_t	aLayersCnt,
		const Int_t *	aLayers )

Definition at line 38 of file TJetNet.cxx.

  : m_Debug (kTRUE)
#else
  : m_Debug (kFALSE)
#endif
{
  // Creates neural network with aLayersCnt number of layers,
  // aTestCount number of patterns for the test set,
  // aTrainCount patterns for the train set.
  // aLayers contains the information for number of the units in the different layers
 
  int i;
  
  if( m_Debug ){ std::cout << "=====> Entering TJetNet::TJetNet(...)" << std::endl; }
 
  m_TestSetCnt  = aTestCount;
  m_TrainSetCnt = aTrainCount;
  m_LayerCount  = aLayersCnt; // Get the number of layers
  
  if( m_LayerCount > 0 )
  {  
    //Perform deep copy of the array holding Layers count
    m_pLayers = new Int_t[ m_LayerCount ];
    for( i = 0; i < m_LayerCount; ++i )
    {
      m_pLayers[ i ] = aLayers[ i ];
    }
  }
 
  m_InputDim = m_pLayers[ 0 ];
  m_OutputDim = m_pLayers[ m_LayerCount - 1 ];
  m_HiddenLayerDim = m_LayerCount-2;
  
 
  m_IsInitialized = kFALSE;
  m_InitLocked = kFALSE;
 
  m_pInputTrainSet  = new TNeuralDataSet( m_TrainSetCnt, GetInputDim() );
  m_pInputTestSet = new TNeuralDataSet( m_TestSetCnt, GetInputDim() );
  m_pOutputTrainSet = new TNeuralDataSet( m_TrainSetCnt, GetOutputDim() );
  m_pOutputTestSet = new TNeuralDataSet( m_TestSetCnt, GetOutputDim() );
  
  m_NormalizeOutput=false;
 
  m_enActFunction=afSigmoid;
 
  SetEpochs( -1 );
 
  if( m_Debug ){ std::cout << "=====> Leaving TJetNet::TJetNet(...)" << std::endl; }
}

~TJetNet()

TJetNet::~TJetNet ( void )

virtual

Definition at line 91 of file TJetNet.cxx.

{
  // Default destructor
  if( m_Debug ){ std::cout << "=====> Entering TJetNet::~TJetNet(...)" << std::endl; }
  delete [] m_pLayers;
  delete m_pInputTestSet;
  delete m_pInputTrainSet;
  delete m_pOutputTestSet;
  delete m_pOutputTrainSet;
  if( m_Debug ){ std::cout << "=====> Leaving TJetNet::~TJetNet(...)" << std::endl; }
}

Member Function Documentation

CopyFile()

Int_t TJetNet::CopyFile ( TString aSrcFile, TString aDestFile )

private

createTrainedNetwork()

TTrainedNetwork * TJetNet::createTrainedNetwork

(

)

const

Definition at line 104 of file TJetNet.cxx.

{
 
  Int_t nInput=GetInputDim();
  Int_t nHidden=GetHiddenLayerDim();
  std::vector<Int_t> nHiddenLayerSize;
  //  Int_t* nHiddenLayerSize=new Int_t[nHidden];
 
  for (Int_t o=0;o<nHidden;++o)
  {
    nHiddenLayerSize.push_back(GetHiddenLayerSize(o+1));
  }
  Int_t nOutput=GetOutputDim();
 
  std::vector<TVectorD*> thresholdVectors;
  std::vector<TMatrixD*> weightMatrices;
 
  for (Int_t o=0;o<nHidden+1;++o)
  {
    int sizeActualLayer=(o<nHidden)?nHiddenLayerSize[o]:nOutput;
    int sizePreviousLayer=(o==0)?nInput:nHiddenLayerSize[o-1];
    thresholdVectors.push_back(new TVectorD(sizeActualLayer));
    weightMatrices.push_back(new TMatrixD(sizePreviousLayer,sizeActualLayer));
  }
 
  for (Int_t o=0;o<nHidden+1;++o)
  {
 
    if (m_Debug)
      if (o<nHidden)
      {
        cout << " Iterating on hidden layer n.: " << o << endl;
      }
      else
      {
        cout << " Considering output layer " << endl;
      }
    
    int sizeActualLayer=(o<nHidden)?nHiddenLayerSize[o]:nOutput;
 
    for (Int_t s=0;s<sizeActualLayer;++s)
    {
      if (o<nHidden)
      {
    if (m_Debug)
          cout << " To hidden node: " << s << endl;
      }
      else
      {
    if (m_Debug)
          cout << " To output node: " << s << endl;
      }
      if (o==0)
      {
    for (Int_t p=0;p<nInput;++p)
    {
      if (m_Debug)
            cout << " W from inp nod: " << p << "weight: " <<
              GetWeight(o+1,s+1,p+1) << endl;
      weightMatrices[o]->operator() (p,s) = GetWeight(o+1,s+1,p+1);
        }
      }
      else
      {
    for (Int_t p=0;p<nHiddenLayerSize[o-1];++p)
    {
      if (m_Debug)
            cout << " W from lay : " << o-1 << " nd: " << 
              p << " weight: " <<
              GetWeight(o+1,s+1,p+1) << endl;
      weightMatrices[o]->operator() (p,s)=GetWeight(o+1,s+1,p+1);
    }
      }
      if (m_Debug)
        cout << " Threshold for node " << s << " : " << 
          GetThreshold(o+1,s+1) << endl;
      thresholdVectors[o]->operator() (s) = GetThreshold(o+1,s+1);
    }
  }
 
  bool linearOutput=false;
  if (this->GetActivationFunction(nHidden)==4) 
  {
    //    cout << " Creating TTrainedNetwork with linear output function" << endl;
    linearOutput=true;
  }
 
  TTrainedNetwork* trainedNetwork=
    new TTrainedNetwork(nInput,
            nHidden,
            nOutput,
            nHiddenLayerSize,
            thresholdVectors,
            weightMatrices,
            m_enActFunction,
                        linearOutput,
                        m_NormalizeOutput);
                        
  return trainedNetwork;
 
}

DumpToFile()

void TJetNet::DumpToFile

(

TString

aFileName = "fort.8"

)

Definition at line 954 of file TJetNet.cxx.

{
  // Dumps the network data into JETNET specific format
  JNDUMP( -8 );
  std::cout << close( 8 ) << std::endl;
  rename( "./fort.8", aFileName );
}

Epoch()

Int_t TJetNet::Epoch

(

void

)

Definition at line 696 of file TJetNet.cxx.

{
  // Initiate one train/test step the network. 
 
  Double_t aTrain, aTest;
  if ( m_CurrentEpoch < m_Epochs )
  {
    m_CurrentEpoch++;
    aTrain = Train();
 
    //      if (m_CurrentEpoch%2)
 
    //      std::cout << " Calls to MSTJN: " << GetMSTJN(6) << 
    //  std::endl;
 
    if ( m_Debug ) 
    {
 
 
      std::cout << "[ " << m_CurrentEpoch << " ] Train: " << aTrain << std::endl;
    }
    if ( ( m_CurrentEpoch % 2 ) == 0 )
    {
      aTest = Test();
      //      if ( m_Debug )
      std::cout << "[" << m_CurrentEpoch << "]: " << GetPARJN(8) << " ";
      std::cout << "Test: " << aTest << std::endl;
    }
  }
  return m_CurrentEpoch;
}

Evaluate() [1/2]

void TJetNet::Evaluate

(

)

Definition at line 926 of file TJetNet.cxx.

{
  //evaluates directly the input provided through SetInputs()
  JNTEST();
}

Evaluate() [2/2]

void TJetNet::Evaluate

(

Int_t

aPattern

)

Definition at line 932 of file TJetNet.cxx.

{
  // Evaluates the network output form the input data specified by the Test Pattern
  for( Int_t i = 0; i < GetInputDim(); i++ )
  {
    JNDAT1.OIN[ i ] = float ( GetInputTestSet( aPattern, i ) );
  }
  JNTEST();
}

GetActivationFunction() [1/2]

Int_t TJetNet::GetActivationFunction

(

Int_t

layerN

)

const

Definition at line 1167 of file TJetNet.cxx.

{ 
  return JNDAT2.IGFN[ layerN ];
}

GetActivationFunction() [2/2]

Int_t TJetNet::GetActivationFunction

(

void

)

const

Definition at line 1161 of file TJetNet.cxx.

{ 
  return JNDAT1.MSTJN[ 2 ]; 
}

GetEpochs()

Int_t TJetNet::GetEpochs ( void )

inline

Definition at line 78 of file TJetNet.h.

{ return m_Epochs; };

GetErrorMeasure()

Int_t TJetNet::GetErrorMeasure

(

void

)

Definition at line 1156 of file TJetNet.cxx.

{ 
  return JNDAT1.MSTJN[ 3 ]; 
}

GetEventWeightTestSet()

Double_t TJetNet::GetEventWeightTestSet ( Int_t aPatternInd )

inline

Definition at line 213 of file TJetNet.h.

{
  return m_pInputTestSet->GetEventWeight( aPatternInd);
}

GetEventWeightTrainSet()

Double_t TJetNet::GetEventWeightTrainSet ( Int_t aPatternInd )

inline

Definition at line 208 of file TJetNet.h.

{
  return m_pInputTrainSet->GetEventWeight( aPatternInd);
}

GetHiddenLayerDim()

Int_t TJetNet::GetHiddenLayerDim ( void ) const

inline

Definition at line 55 of file TJetNet.h.

{ return mHiddenLayerDim; };

GetHiddenLayerSize()

Int_t TJetNet::GetHiddenLayerSize ( Int_t number ) const

inline

Definition at line 56 of file TJetNet.h.

{ return m_pLayers[ number ]; };

GetInitialWeightsWidth()

Double_t TJetNet::GetInitialWeightsWidth

(

void

)

Definition at line 1187 of file TJetNet.cxx.

{ 
  return JNDAT1.PARJN[ 3 ]; 
}

GetInputDim()

Int_t TJetNet::GetInputDim ( void ) const

inline

Definition at line 54 of file TJetNet.h.

{ return m_pLayers[ 0 ]; };

GetInputTestSet()

Double_t TJetNet::GetInputTestSet	(	Int_t	aPatternInd,
		Int_t	aInputInd )

Definition at line 763 of file TJetNet.cxx.

{
  // Returns the value of the cell corresponding to unit aInputInd in pattern aPatternInd into INPUT TEST set
  return m_pInputTestSet->GetData( aPatternInd, aInputInd );
}

GetInputTrainSet()

Double_t TJetNet::GetInputTrainSet ( Int_t aPatternInd, Int_t aInputInd )

inline

Definition at line 202 of file TJetNet.h.

{
  // Returns the value of the cell corresponding to unit aInputInd in pattern aPatternInd into INPUT TRAIN set
  return m_pInputTrainSet->GetData( aPatternInd, aInputInd );
}

GetLayerCount()

Int_t TJetNet::GetLayerCount ( void )

inline

Definition at line 105 of file TJetNet.h.

{ return m_LayerCount; };

GetLearningRate()

Double_t TJetNet::GetLearningRate

(

void

)

Definition at line 1177 of file TJetNet.cxx.

{ 
  return JNDAT1.PARJN[ 0 ]; 
}

GetLearningRateDecrease()

Double_t TJetNet::GetLearningRateDecrease

(

void

)

Definition at line 1192 of file TJetNet.cxx.

{ 
  return JNDAT1.PARJN[ 10 ]; 
}

GetMomentum()

Double_t TJetNet::GetMomentum

(

void

)

Definition at line 1182 of file TJetNet.cxx.

{ 
  return JNDAT1.PARJN[ 1 ]; 
}

GetMSTJN()

Int_t TJetNet::GetMSTJN

(

Int_t

aIndex

)

Definition at line 1202 of file TJetNet.cxx.

{
  return JNDAT1.MSTJN[ aIndex ]; 
}

GetOutput()

Double_t TJetNet::GetOutput

(

Int_t

aIndex = 0

)

Definition at line 948 of file TJetNet.cxx.

{
  // Returns the output of the network 
  return Double_t ( JNDAT1.OUT[ aIndex ] );
}

GetOutputDim()

Int_t TJetNet::GetOutputDim ( void ) const

inline

Definition at line 57 of file TJetNet.h.

{ return m_pLayers[ m_LayerCount - 1 ]; };

GetOutputTestSet()

Double_t TJetNet::GetOutputTestSet	(	Int_t	aPatternInd,
		Int_t	aOutputInd )

Definition at line 769 of file TJetNet.cxx.

{
  // Returns the value of the cell corresponding to unit aInputInd in pattern aPatternInd into OUTPUT TEST set
  return m_pOutputTestSet->GetData( aPatternInd, aOutputInd );
}

GetOutputTrainSet()

Double_t TJetNet::GetOutputTrainSet	(	Int_t	aPatternInd,
		Int_t	aOutputInd )

Definition at line 746 of file TJetNet.cxx.

{
  // Returns the value of the cell corresponding to unit aInputInd in pattern aPatternInd into OUTPUT TRAIN set
  return m_pOutputTrainSet->GetData( aPatternInd, aOutputInd );
}

GetPARJN()

Double_t TJetNet::GetPARJN

(

Int_t

aIndex

)

Definition at line 1207 of file TJetNet.cxx.

{
  return JNDAT1.PARJN[ aIndex ];
}

GetPatternsPerUpdate()

Int_t TJetNet::GetPatternsPerUpdate

(

void

)

Definition at line 1172 of file TJetNet.cxx.

{ 
  return JNDAT1.MSTJN[ 1 ]; 
}

GetPottsUnits()

Int_t TJetNet::GetPottsUnits

(

)

Definition at line 1197 of file TJetNet.cxx.

{
  return JNINT2.IPOTT;
} 

GetTestSetCnt()

Int_t TJetNet::GetTestSetCnt ( void ) const

inline

Definition at line 53 of file TJetNet.h.

{ return m_TestSetCnt; };

GetThreshold()

Double_t TJetNet::GetThreshold	(	Int_t	aLayerInd,
		Int_t	aNodeInd ) const

Definition at line 979 of file TJetNet.cxx.

{
  //Returns the node threshold in the specific layer
  return Double_t ( JNINT1.T[ JNINDX( aLayerInd, aNodeInd, 0 )-1 ] );
  //GP: ONE HAS TO PAY ATTENTION TO THIS STUPID -1!!!
}

GetTrainSetCnt()

Int_t TJetNet::GetTrainSetCnt ( void ) const

inline

Definition at line 52 of file TJetNet.h.

{ return m_TrainSetCnt; };

GetUnitCount()

Int_t TJetNet::GetUnitCount

(

Int_t

aLayer

)

Definition at line 1064 of file TJetNet.cxx.

{ 
  // Returns the number of the units in specfic layer
  if( ( aLayer > -1 ) && ( aLayer < m_LayerCount ) ) 
    return JNDAT1.MSTJN[ 9 + aLayer ]; 
}

GetUpdatesPerEpoch()

Int_t TJetNet::GetUpdatesPerEpoch

(

void

)

Definition at line 1146 of file TJetNet.cxx.

{ 
  return JNDAT1.MSTJN[ 8 ]; 
}

GetUpdatingProcedure()

Int_t TJetNet::GetUpdatingProcedure

(

void

)

Definition at line 1151 of file TJetNet.cxx.

{  
  return JNDAT1.MSTJN[ 3 ]; 
}

GetWeight()

Double_t TJetNet::GetWeight	(	Int_t	aLayerInd,
		Int_t	aNodeInd,
		Int_t	aConnectedNodeInd ) const

Definition at line 972 of file TJetNet.cxx.

{
  // Returns the node weight in specific Layer
  return Double_t ( JNINT1.W[ JNINDX( aLayerInd, aNodeInd, aConnectedNodeInd )-1 ] );
  //GP: ONE HAS TO PAY ATTENTION TO THIS STUPID -1!!!
}

Init()

void TJetNet::Init

(

void

)

Definition at line 670 of file TJetNet.cxx.

{
  // Initializes the neuaral network
  Int_t i;
  JNDAT1.MSTJN[ 0 ] = m_LayerCount; // Set the number of layers
 
  // Set the number of nodes for each layer
  for( i = 0; i < m_LayerCount; i++ )
  {
    if ( m_Debug ) std::cout << "Layer " << i + 1 << " has " << m_pLayers[ i ] << " units." << std::endl; 
    JNDAT1.MSTJN[ 9 + i ] = m_pLayers[ i ]; 
  }
   
  cout << " calling JNINIT " << endl;
  JNINIT();
 
  if (m_NormalizeOutput)  
  {
    std::cout << " Setting to normalize output nodes: POTT nodes " << std::endl;
    SetPottsUnits(m_pLayers[m_LayerCount-1]); 
  }
  
  cout << " finishing calling JNINIT " << endl;
  m_IsInitialized = kTRUE;
}

LoadDataAscii()

void TJetNet::LoadDataAscii

(

TString

aFileName = "jndata.dat"

)

Definition at line 815 of file TJetNet.cxx.

{
  // Loads the input/output test/train data from plain text file 
  ifstream in;
  int i, j, k, l, m;
  int aiParam[ 5 ];//iTrainCount, iTestCount, iInputDim, iHiddenDim, iOutputDim;
  Bool_t bFlag;
  Double_t tmp;
  Int_t iPatternLength;
 
  in.open( aFileName );
  bFlag = Bool_t( in.is_open() );
  if ( in )
  { 
    in >> m_LayerCount;
    if( m_Debug ){ std::cout << "Layers Count Set to " << m_LayerCount << std::endl;}
    i = 0;
 
    delete [] m_pLayers;
    m_pLayers = new Int_t[ m_LayerCount ];
 
    if( m_Debug ){ std::cout << "Updating the Layers Nodes Counters..." << std::endl; }
    while( ( i < m_LayerCount ) && ( !in.eof() ) )
    {
      in >> m_pLayers[ i ];
      if( m_Debug ){ std::cout << "Layer [ " << i + 1 << " ] has " << m_pLayers[ i ] << " units" << std::endl; }
      i++;
    }
 
    m_InputDim = m_pLayers[ 0 ];
    m_OutputDim = m_pLayers[ m_LayerCount - 1 ];
    m_HiddenLayerDim = m_LayerCount-2;
 
    //Get the patterns count per line 
    iPatternLength = m_InputDim + m_OutputDim;
    if( m_Debug ){ std::cout << "Patterns per line = " << iPatternLength << std::endl; } 
    in >> m_TrainSetCnt;
    if( m_Debug ){ std::cout << "Train Set has " << m_TrainSetCnt << " patterns." << std::endl; }
    in >> m_TestSetCnt;
    if( m_Debug ){ std::cout << "Test Set has " << m_TestSetCnt << " patterns." << std::endl; }
    
    delete m_pInputTestSet;
    delete m_pInputTrainSet;
    delete m_pOutputTestSet;
    delete m_pOutputTrainSet;
    
    m_pInputTrainSet  = new TNeuralDataSet( m_TrainSetCnt, GetInputDim() );
    m_pInputTestSet   = new TNeuralDataSet( m_TestSetCnt, GetInputDim() );
    m_pOutputTrainSet = new TNeuralDataSet( m_TrainSetCnt, GetOutputDim() );
    m_pOutputTestSet  = new TNeuralDataSet( m_TestSetCnt, GetOutputDim() );
 
    i = 0;
    j = 0;
    
    while( ( i < ( m_TrainSetCnt + m_TestSetCnt ) ) && ( !in.eof() ) )
    {
      j = 0;
      while( ( j < iPatternLength ) && ( !in.eof() ) )
      {
    if( i < m_TrainSetCnt )
    {
      if( j < m_InputDim )
      {
        //Train Input Set
        in >> tmp;
        SetInputTrainSet( i, j, tmp );
      }
      else 
      {
        //Train Output Set 
        m = j - m_InputDim;
        in >> tmp;
        SetOutputTrainSet( i, m, tmp );
      }
    }
    else
    {
      l = i - m_TrainSetCnt;
      if( j < m_InputDim )
        {
          //Test Input Set
          in >> tmp;
          SetInputTestSet( l, j, tmp );
        }
          else
        {
          //Test Output Set
          m = j - m_InputDim;
          in >> tmp;
          SetOutputTestSet( l, m, tmp );
        }
 
    }    
    j++;
      }
      i++;
    }
  }
  in.close();
}

LoadDataRoot()

void TJetNet::LoadDataRoot

(

TString

aFileName = "jndata.root"

)

Definition at line 921 of file TJetNet.cxx.

{
  // Loads the neural network from ROOT file
}

LockInit()

void TJetNet::LockInit ( void )

inline

Definition at line 137 of file TJetNet.h.

{ m_InitLocked = kTRUE; };

mSetThreshold()

void TJetNet::mSetThreshold ( Double_t threshold, Int_t aLayerInd, Int_t aNodeInd )

private

Definition at line 290 of file TJetNet.cxx.

{
  JNINT1.T[ JNINDX( aLayerInd, aNodeInd, 0 )-1 ]=threshold;
}

mSetWeight()

void TJetNet::mSetWeight ( Double_t weight, Int_t aLayerInd, Int_t aNodeInd, Int_t aConnectedNodeInd )

private

Definition at line 285 of file TJetNet.cxx.

{
  JNINT1.W[ JNINDX( aLayerInd, aNodeInd, aConnectedNodeInd )-1 ]=weight;
}

Normalize()

void TJetNet::Normalize

(

void

)

Definition at line 1048 of file TJetNet.cxx.

{
  // Normilizes Inputs (both test and train)
  m_pInputTrainSet->Normalize();
  m_pInputTestSet->Normalize();
}

NormalizeOutput()

void TJetNet::NormalizeOutput

(

bool

isTrue = true

)

Definition at line 1141 of file TJetNet.cxx.

{
  m_NormalizeOutput=isTrue;
}

Print()

void TJetNet::Print

(

void

)

Definition at line 295 of file TJetNet.cxx.

{
  // Prints on the screen, information for the neural network
  Int_t i;
 
  std::cout << "TJetNet" << std::endl;
  std::cout << "Number of layers: " << m_LayerCount << std::endl;
 
  for( i = 0; i < m_LayerCount; i++ )
  {
    std::cout << "\t\tNumber of units in layer " << i << " : " <<  m_pLayers[ i ] << std::endl;
  }
 
  std::cout << "Epochs: " << GetEpochs() << std::endl;
  std::cout << "Updates Per Epoch: " << GetUpdatesPerEpoch() << std::endl;
  std::cout << "Updating Procedure: " << GetUpdatingProcedure() << std::endl;
  std::cout << "Error Measure: " << GetErrorMeasure() << std::endl;
  std::cout << "Patterns Per Update: " << GetPatternsPerUpdate() << std::endl;
  std::cout << "Learning Rate: " << GetLearningRate() << std::endl;
  std::cout << "Momentum: " << GetMomentum() << std::endl;
  std::cout << "Initial Weights Width: " << GetInitialWeightsWidth() << std::endl;
  std::cout << "Learning Rate Decrease: " << GetLearningRateDecrease() << std::endl;
  std::cout << "Activation Function: " << GetActivationFunction() << std::endl;
}

Randomize()

void TJetNet::Randomize

(

void

)

Definition at line 1055 of file TJetNet.cxx.

{
  // Randomizes Inputs and Outputs of both train and test sets
  m_pInputTrainSet->Randomize();
  m_pInputTestSet->Randomize();
  m_pOutputTrainSet->Randomize();
  m_pOutputTestSet->Randomize();
}

readBackTrainedNetwork()

void TJetNet::readBackTrainedNetwork

(

const TTrainedNetwork *

trainedNetwork

)

Definition at line 207 of file TJetNet.cxx.

{
 
  Int_t nInput=GetInputDim();
  Int_t nHidden=GetHiddenLayerDim();
  std::vector<Int_t> nHiddenLayerSize;
 
  if (trainedNetwork->getnHidden()!=nHidden)
  {
    cout << " Network doesn't match.. not loading.." << endl;
    return;
  }
 
  for (Int_t o=0;o<nHidden;++o)
  {
    nHiddenLayerSize.push_back(GetHiddenLayerSize(o+1));
    if (nHiddenLayerSize[o]!=trainedNetwork->getnHiddenLayerSize()[o])
    {
      cout << " Network doesn't match... not loading..." << endl;
      return;
    }
  }
  Int_t nOutput=GetOutputDim();
 
  if (trainedNetwork->getnInput()!=nInput)
  {
    cout << " Network doesn't match... not loading.." << endl;
    return;
  }
 
 
  if (trainedNetwork->getnOutput()!=nOutput)
  {
    cout << " Network doesn't match.. not loading.." << endl;
    return;
  }
  
  //OK, everything matches... can go on...
  
  std::vector<TVectorD*> thresholdVectors=trainedNetwork->getThresholdVectors();
  std::vector<TMatrixD*> weightMatrices=trainedNetwork->weightMatrices();
  //ownership remains of the TTrainedNetwork
 
  for (Int_t o=0;o<nHidden+1;++o)
  {
    int sizeActualLayer=(o<nHidden)?nHiddenLayerSize[o]:nOutput;
    int sizePreviousLayer=(o==0)?nInput:nHiddenLayerSize[o-1];
  
    for (Int_t s=0;s<sizeActualLayer;++s)
    {
      Double_t nodeValue=0.;
      if (o==0)
      {
    for (Int_t p=0;p<nInput;++p)
    {
      mSetWeight(weightMatrices[o]->operator() (p,s),o+1,s+1,p+1);
        }
      }
      else
      {
    for (Int_t p=0;p<nHiddenLayerSize[o-1];++p)
    {
      mSetWeight(weightMatrices[o]->operator() (p,s),o+1,s+1,p+1);
    }
      }
      mSetThreshold(thresholdVectors[o]->operator() (s),o+1,s+1);
    }
  }     
  if (trainedNetwork->getIfLinearOutput()==true)
  {
    cout << " Setting linear output function " << endl;
    this->SetActivationFunction(nHidden,4);
  }
  
  cout << " Successfully read back Trained Network " << endl;
}

ReadFromFile()

void TJetNet::ReadFromFile

(

TString

aFileName = "fort.8"

)

Definition at line 962 of file TJetNet.cxx.

{
  // Loads the network from JETNET specific file
  rename( aFileName, "./fort.12" );
  JNREAD( -12 );
  Reinitialize();
  rename( "./fort.12", aFileName );
  //std::cout << close( 12 ) << std::endl;
}

Reinitialize()

void TJetNet::Reinitialize ( void )

private

Definition at line 1020 of file TJetNet.cxx.

{
  //Initializes the settings of the network
    Int_t i;
    
   m_LayerCount = JNDAT1.MSTJN[ 0 ]; // Set the number of layers
   
   delete [] m_pLayers;
   m_pLayers = new Int_t[ m_LayerCount ];
   
  // Set the number of nodes for each layer
  for( i = 0; i < m_LayerCount; i++ )
  {
    m_pLayers[ i ] = JNDAT1.MSTJN[ 9 + i ]; 
  }
  
  m_pInputTrainSet  = new TNeuralDataSet( m_TrainSetCnt, GetInputDim() );
  m_pInputTestSet   = new TNeuralDataSet( m_TestSetCnt, GetInputDim() );
  m_pOutputTrainSet = new TNeuralDataSet( m_TrainSetCnt, GetOutputDim() );
  m_pOutputTestSet  = new TNeuralDataSet( m_TestSetCnt, GetOutputDim() );
 
  m_InputDim = m_pLayers[ 0 ];
  m_OutputDim = m_pLayers[ m_LayerCount - 1 ];
  m_HiddenLayerDim = m_LayerCount-2;
 
 
}  

SaveDataAscii()

void TJetNet::SaveDataAscii

(

TString

aFileName = "jndata.dat"

)

Definition at line 775 of file TJetNet.cxx.

{
  // Saves the Input/Output test and train data in plain text file
  ofstream out;
  int i, j;
 
  // Open ASCII file
  out.open( aFileName );
 
  //Write the number of layers, including the input and output
  out << m_LayerCount << std::endl;
 
  // Write into the file the number of units in input, hidden and output layers  
  for ( i = 0; i < m_LayerCount; i++ ) out << m_pLayers[ i ] << " ";
  out << std::endl;
 
  // Write the size of Train and Test sets 
  out << m_TrainSetCnt << " " << m_TestSetCnt << std::endl;
 
  // Dump the Train set : Input1 Input2 ... InputN Output1 Output2 ... OutputN
  for ( i = 0; i < m_TrainSetCnt; i++ )
  {
    out << GetInputTrainSet( i, 0 );
    for( j = 1; j < m_pLayers[ 0 ]; j++ ) out << " " << GetInputTrainSet( i, j );
    for( j = 0; j < m_pLayers[ m_LayerCount - 1 ]; j++ ) out << " " << GetOutputTrainSet( i, j );
    out << std::endl;
  }
 
 // Dump the Test set : Input1 Input2 ... InputN Output1 Output2 ... OutputN
  for ( i = 0; i < m_TestSetCnt; i++ )
  {
    out << GetInputTestSet( i, 0 );
    for( j = 1; j < m_pLayers[ 0 ]; j++ ) out << " " << GetInputTestSet( i, j );
    for( j = 0; j < m_pLayers[ m_LayerCount - 1 ]; j++ ) out << " " << GetOutputTestSet( i, j );
    out << std::endl;
  }
  // Close the file
  out.close();
}

SaveDataRoot()

void TJetNet::SaveDataRoot

(

TString

aFileName = "jndata.root"

)

Definition at line 916 of file TJetNet.cxx.

{
  // Saves the neural network in ROOT file
}

SelectiveFields()

void TJetNet::SelectiveFields	(	Int_t	aLayerA,
		Int_t	aNodeA1,
		Int_t	aNodeA2,
		Int_t	aNodeB1,
		Int_t	aNodeB2,
		Int_t	aSwitch = 0 )

Definition at line 986 of file TJetNet.cxx.

{
  // JetNet Selective Fields
  Int_t tmp, i1, i2, j1, j2;
 
  if( ( aLayerA > 0 ) && ( aLayerA < m_LayerCount ) )
  {
    i1 = TMath::Abs( aNodeA1 ); 
    i2 = TMath::Abs( aNodeA2 );
    j1 = TMath::Abs( aNodeB1 ); 
    j2 = TMath::Abs( aNodeB2 );
 
    if( i1 > i2 )
    {
      tmp = i1;
      i1 = i2;
      i2 = i1;
    }//if
 
    if( i1 > i2 )
    {
      tmp = i1;
      i1 = i2;
      i2 = i1;
    }//if
 
    if( ( i1 < m_pLayers[ aLayerA ] ) && ( i2 < m_pLayers[ aLayerA ] ) && 
    ( j1 < m_pLayers[ aLayerA - 1 ] ) && ( j2 < m_pLayers[ aLayerA - 1 ] ) )
    {
      JNSEFI( aLayerA, i1, i2, j1, j2, aSwitch );
    }//if
  } //if
}

SetActivationFunction() [1/2]

void TJetNet::SetActivationFunction

(

Int_t

aValue

)

Definition at line 1091 of file TJetNet.cxx.

{ 
  // Set the kind of activation function used
  JNDAT1.MSTJN[ 2 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetActivationFunction() [2/2]

void TJetNet::SetActivationFunction	(	Int_t	aValue,
		Int_t	layerN )

Definition at line 1098 of file TJetNet.cxx.

{ 
  // Set the kind of activation function used
  JNDAT2.IGFN[ layerN ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetEpochs()

void TJetNet::SetEpochs ( const Int_t aEpochs )

inline

Definition at line 79 of file TJetNet.h.

{ m_Epochs = aEpochs; m_CurrentEpoch = 0; };

SetErrorMeasure()

void TJetNet::SetErrorMeasure

(

Int_t

aValue

)

Definition at line 1085 of file TJetNet.cxx.

{  
  JNDAT1.MSTJN[ 3 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetEventWeightTestSet()

void TJetNet::SetEventWeightTestSet	(	Int_t	aPatternInd,
		Double_t	aValue )

Definition at line 758 of file TJetNet.cxx.

{
  m_pInputTestSet->SetEventWeight(aPatternInd,aValue);
}

SetEventWeightTrainSet()

void TJetNet::SetEventWeightTrainSet	(	Int_t	aPatternInd,
		Double_t	aValue )

Definition at line 752 of file TJetNet.cxx.

{
  m_pInputTrainSet->SetEventWeight(aPatternInd,aValue);
}

SetInitialWeightsWidth()

void TJetNet::SetInitialWeightsWidth

(

Double_t

aValue

)

Definition at line 1124 of file TJetNet.cxx.

{ 
  JNDAT1.PARJN[ 3 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetInputs()

void TJetNet::SetInputs	(	Int_t	aIndex = 0,
		Double_t	aValue = 0.0 )

Definition at line 942 of file TJetNet.cxx.

{
  // Directly sets the inputs of the network 
  JNDAT1.OIN[ aIndex ] = float ( aValue );
}

SetInputTestSet()

void TJetNet::SetInputTestSet	(	Int_t	aPatternInd,
		Int_t	aInputInd,
		Double_t	aValue )

Definition at line 740 of file TJetNet.cxx.

{
  // Changes the value of the cell corresponding to unit aInputInd in pattern aPatternInd into INPUT TEST set
  m_pInputTestSet->SetData( aPatternInd, aInputInd, aValue );
}

SetInputTrainSet()

void TJetNet::SetInputTrainSet	(	Int_t	aPatternInd,
		Int_t	aInputInd,
		Double_t	aValue )

Definition at line 728 of file TJetNet.cxx.

{
  // Changes the value of the cell corresponding to unit aInputInd in pattern aPatternInd into INPUT TRAIN set
  m_pInputTrainSet->SetData( aPatternInd, aInputInd, aValue );
}

SetLearningRate()

void TJetNet::SetLearningRate

(

Double_t

aValue

)

Definition at line 1111 of file TJetNet.cxx.

{ 
  // Change the Learning Rate
  JNDAT1.PARJN[ 0 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetLearningRateDecrease()

void TJetNet::SetLearningRateDecrease

(

Double_t

aValue

)

Definition at line 1130 of file TJetNet.cxx.

{ 
  JNDAT1.PARJN[ 10 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetMomentum()

void TJetNet::SetMomentum

(

Double_t

aValue

)

Definition at line 1118 of file TJetNet.cxx.

{ 
  JNDAT1.PARJN[ 1 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetMSTJN()

void TJetNet::SetMSTJN	(	Int_t	aIndex,
		Int_t	aValue )

Definition at line 1212 of file TJetNet.cxx.

{
  JNDAT1.MSTJN[ aIndex ] = aValue;
}

SetOutputTestSet()

void TJetNet::SetOutputTestSet ( Int_t aPatternInd, Int_t aOutputInd, Double_t aValue )

inline

Definition at line 196 of file TJetNet.h.

{
  // Changes the value of the cell corresponding to unit aInputInd in pattern aPatternInd into OUTPUT TEST set
  m_pOutputTestSet->SetData( aPatternInd, aOutputInd, aValue );
}

SetOutputTrainSet()

void TJetNet::SetOutputTrainSet	(	Int_t	aPatternInd,
		Int_t	aOutputInd,
		Double_t	aValue )

Definition at line 734 of file TJetNet.cxx.

{
  // Changes the value of the cell corresponding to unit aInputInd in pattern aPatternInd into OUTPUT TRAIN set
  m_pOutputTrainSet->SetData( aPatternInd, aOutputInd, aValue );
}

SetPARJN()

void TJetNet::SetPARJN	(	Int_t	aIndex,
		Double_t	aValue )

Definition at line 1217 of file TJetNet.cxx.

{
  JNDAT1.PARJN[ aIndex ] = aValue;
}

SetPatternsPerUpdate()

void TJetNet::SetPatternsPerUpdate

(

Int_t

aValue

)

Definition at line 1105 of file TJetNet.cxx.

{ 
  JNDAT1.MSTJN[ 1 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetPottsUnits()

void TJetNet::SetPottsUnits

(

Int_t

aValue

)

Definition at line 1136 of file TJetNet.cxx.

{
  JNINT2.IPOTT = aValue; 
} 

SetUpdatesPerEpoch()

void TJetNet::SetUpdatesPerEpoch

(

Int_t

aValue

)

Definition at line 1071 of file TJetNet.cxx.

{ 
  // Sets the number of the updates per epoch
  JNDAT1.MSTJN[ 8 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

SetUpdatingProcedure()

void TJetNet::SetUpdatingProcedure

(

Int_t

aValue

)

Definition at line 1078 of file TJetNet.cxx.

{  
  // Set specific weights update function
  JNDAT1.MSTJN[ 4 ] = aValue; 
  if( !m_InitLocked ) this->Init();
}

Shuffle()

void TJetNet::Shuffle	(	Bool_t	aShuffleTrainSet = true,
		Bool_t	aShuffleTestSet = true )

Definition at line 1222 of file TJetNet.cxx.

{
  // Shuffles the train and/or test input/output sets
  TTimeStamp ts;
  Int_t Seed = ts.GetSec();
  if ( aShuffleTrainSet )
  {
    
    m_pInputTrainSet->Shuffle( Seed );
    m_pOutputTrainSet->Shuffle( Seed );
  }
  //Shuffle Test Set
  if ( aShuffleTestSet )
  {
    Seed = ts.GetSec();
    m_pInputTestSet->Shuffle( Seed );
    m_pOutputTestSet->Shuffle( Seed );
  }
 
  return;
} 

Test()

Double_t TJetNet::Test

(

void

)

Definition at line 320 of file TJetNet.cxx.

{
  // Initiate test cycle of the neural network
  Int_t NRight = 0;
  Double_t fMeanError = 0.0;
  Double_t *TMP;
  Int_t  NPatterns = GetTestSetCnt();
 
 
  for( Int_t iPattern = 0; iPattern < NPatterns; iPattern++ )
  {
 
    for( Int_t i = 0; i < GetInputDim(); i++ )
    {
      JNDAT1.OIN[ i ] = float ( GetInputTestSet( iPattern, i ) );
    }
 
    NWJNWGT.OWGT = GetEventWeightTestSet( iPattern );
 
    JNTEST();
 
    for( Int_t j = 0; j < GetOutputDim(); j++ )
    {
      fMeanError+= NWJNWGT.OWGT * 
        std::pow(JNDAT1.OUT[ j ]-float( GetOutputTestSet( iPattern, j )),2)/(float)GetOutputDim();
    }
 
 
 
    if( m_Debug ) std::cout << "Testing [ " << iPattern << " ] - "  << JNDAT1.OIN[ 0 ] 
                           << " => " << JNDAT1.OUT[ 0 ] << std::endl;
 
  }
 
  fMeanError/=2.*NPatterns;
 
  if (m_Debug)
    std::cout << " Test error: " << fMeanError << endl;
 
  return fMeanError;
}

TestBTAG()

Double_t TJetNet::TestBTAG

(

void

)

Definition at line 362 of file TJetNet.cxx.

{
 
  bool test=false;
 
  // Initiate test cycle of the neural network
  Int_t NRight = 0;
  Double_t fMeanError = 0.0;
  Double_t *TMP;
  Int_t  NPatterns = GetTestSetCnt();
  if (test)
  {
    NPatterns = GetTrainSetCnt();
  }
 
  vector<double> eff;
  eff.push_back(0.5);
  eff.push_back(0.6);
  eff.push_back(0.7);
 
  //test also the b-tagging performance directly during testing !!!
  vector<TH1F*> histoEfficienciesC;
  vector<TH1F*> histoEfficienciesL;
  TString histoEffStringC("histoEffC");
  TString histoEffStringL("histoEffL");
  for (int i=0;i<GetOutputDim();i++)
  {
    TString string1=histoEffStringC;
    string1+=i;
    TH1F* histo=new TH1F(string1,
                         string1,
                         20000,
                         -2,3);
    TString string2=histoEffStringL;
    string2+=i;
    TH1F* histo2=new TH1F(string2,
                          string2,
                          20000,
                          -2,3);
    histoEfficienciesC.push_back(histo);
    histoEfficienciesL.push_back(histo2);
  }
 
  for( Int_t iPattern = 0; iPattern < NPatterns; iPattern++ )
  {
 
    for( Int_t i = 0; i < GetInputDim(); i++ )
    {
      if (!test)
      {
        JNDAT1.OIN[ i ] = float ( GetInputTestSet( iPattern, i ) );
        NWJNWGT.OWGT = GetEventWeightTestSet( iPattern );
      }
      else
      {
        JNDAT1.OIN[ i ] = float (GetInputTrainSet( iPattern, i ) );
        NWJNWGT.OWGT = GetEventWeightTrainSet( iPattern );
      }
 
    }
 
    JNTEST();
 
    int active=0;
    for( Int_t j = 0; j < GetOutputDim(); j++ )
    {
      if (!test)
      {
        fMeanError+= NWJNWGT.OWGT * 
          std::pow(JNDAT1.OUT[ j ]-float( GetOutputTestSet( iPattern, j )),2)/(float)GetOutputDim();
      }
      else
      {
        fMeanError+= NWJNWGT.OWGT * 
          std::pow(JNDAT1.OUT[ j ]-float( GetOutputTrainSet( iPattern, j )),2)/(float)GetOutputDim();
      }
        
 
      //        std::cout << " j " <<  j << " is " << GetOutputTestSet( iPattern, j) << std::endl;
 
      if (!test)
      {
        if (fabs(float( GetOutputTestSet( iPattern, j)) - 1) < 1e-4)
        {
          active = j;
        }
      }
      else
      {
        if (fabs(float( GetOutputTrainSet( iPattern, j)) - 1) < 1e-4)
        {
          active = j;
        }
      }
    }
 
    //      if (m_Debug) std::cout << " active is: " << active << std::endl;
 
    //      if (m_Debug) std::cout << " filling histograms " << std::endl;
 
    if (JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 1 ] >= 0)
    {
      histoEfficienciesC[active]->Fill( JNDAT1.OUT[ 0 ] / 
                                        ( JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 1 ]),
                                        NWJNWGT.OWGT);
                                                                                
      //        if( m_Debug ) std::cout << "Filled: " << JNDAT1.OUT[ 0 ] / 
      //                         ( JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 2 ]) << std::endl;
      
    }
    else
    {
      std::cout << " Filled 0 " << std::endl;
      histoEfficienciesC[active]->Fill( 0 );
    }
      
        
    if (JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 2 ] >= 0)
    {
      histoEfficienciesL[active]->Fill( JNDAT1.OUT[ 0 ] / 
                                        ( JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 2 ]),
                                        NWJNWGT.OWGT);
      //        if( m_Debug ) std::cout << "Filled: " << JNDAT1.OUT[ 0 ] / 
      //                         ( JNDAT1.OUT[ 0 ] + JNDAT1.OUT[ 1 ]) << std::endl;
      
    }
    else
    {
      std::cout << " Filled 0 " << std::endl;
      histoEfficienciesL[active]->Fill( 0 );
    }
 
    if( m_Debug ) std::cout << "Testing [ " << iPattern << " ] - "  << JNDAT1.OIN[ 0 ] 
                           << " => " << JNDAT1.OUT[ 0 ] << std::endl;
      
  }// finish patterns
  
  if (m_Debug) std::cout << " Finished patterns... " << std::endl;
  
  TFile* newFile=new TFile("test.root","recreate");
  histoEfficienciesL[0]->Write();
  histoEfficienciesL[1]->Write();
  histoEfficienciesL[2]->Write();
  histoEfficienciesC[0]->Write();
  histoEfficienciesC[1]->Write();
  histoEfficienciesC[2]->Write();
  newFile->Write();
  newFile->Close();
 
  //for C-jet rejection
  
  for (int u=0;u<2;u++)
  {
    vector<TH1F*>* myVectorHistos;
    if (u==0) 
    {
      std::cout << "c-rej --> ";
      myVectorHistos=&histoEfficienciesC;
    }
    if (u==1)
    {
      std::cout << "l-rej --> ";
      myVectorHistos=&histoEfficienciesL;
    }
    
      
    if (m_Debug) std::cout << " 1 " << std::endl;
    
    Double_t allb=(*myVectorHistos)[0]->GetEntries();
    Double_t allc=(*myVectorHistos)[1]->GetEntries();
    Double_t allu=(*myVectorHistos)[2]->GetEntries();
    
    if (m_Debug) std::cout << " allb " << allb << std::endl;
 
    Double_t allbsofar=0;
    
    vector<int> binN_Eff;
    vector<bool> ok_eff;
    
    for (int r=0;r<eff.size();r++)
    {
      ok_eff.push_back(false);
      binN_Eff.push_back(0);
    }
 
    for (int s=0;s<(*myVectorHistos)[0]->GetNbinsX()+1;s++) {
      allbsofar+=(*myVectorHistos)[0]->GetBinContent((*myVectorHistos)[0]->GetNbinsX()+1-s);
      bool nothingMore(true);
 
 
      for (int r=0;r<eff.size();r++)
      {
        if (m_Debug) std::cout << " actual eff: " << allbsofar / allb << std::endl;
 
        if ((!ok_eff[r]) && allbsofar / allb > eff[r])
        {
          binN_Eff[r]=s;
          ok_eff[r]=true;
          if (m_Debug) std::cout << " bin: " << s << " eff: " << allbsofar / allb << std::endl;
          //          std::cout << " Cut value: " << (*myVectorHistos)[0]->GetBinCenter(s) << std::endl;
        }
        else if (allbsofar / allb <= eff[r])
        {
          nothingMore=false;
        }
      }
      if (nothingMore) break;
    }
    
    
    for (int r=0;r<eff.size();r++)
    {
      
      std::cout << " " << eff[r];
 
      std::cout << " check: " << (double)(*myVectorHistos)[0]->Integral((*myVectorHistos)[0]->GetNbinsX()-binN_Eff[r],
                                                                        (*myVectorHistos)[1]->GetNbinsX()+1)
        / (double)allb;
      
      double effc=(*myVectorHistos)[1]->Integral((*myVectorHistos)[0]->GetNbinsX()-binN_Eff[r],
                                                 (*myVectorHistos)[1]->GetNbinsX()+1);
      effc /= allc;
      double effl=(*myVectorHistos)[2]->Integral((*myVectorHistos)[0]->GetNbinsX()-binN_Eff[r],
                                                 (*myVectorHistos)[2]->GetNbinsX()+1);
      effl /= allu;
      
      if (effc!=0)
      {
        std::cout << " c: " << 1/effc;
      }
      if (effl!=0)
      {
        std::cout << " l: " << 1/effl;
      }
      
    }
    std::cout << std::endl;
  }
  
  for( Int_t j = 0; j < GetOutputDim(); j++ )
  {
    delete histoEfficienciesC[j];
    delete histoEfficienciesL[j];
  }
   
 
  fMeanError/=2.*NPatterns;
 
  if (m_Debug)
    std::cout << " Test error: " << fMeanError << endl;
 
  return fMeanError;
}

Train()

Double_t TJetNet::Train

(

void

)

Definition at line 618 of file TJetNet.cxx.

{
  // Initiate the train phase for the neural network
  Int_t NRight = 0;
  Double_t fMeanError = 0.0;
  Int_t  NPatterns = GetTrainSetCnt();
 
  //  cout << " NPatterns is: " << NPatterns << endl;
 
  Int_t inputDim=GetInputDim();
  Int_t outputDim=GetOutputDim();
  Int_t updatesPerEpoch=GetUpdatesPerEpoch();
  Int_t patternsPerUpdate=GetPatternsPerUpdate();
  
  if (updatesPerEpoch*patternsPerUpdate<1./2.*NPatterns) 
  {
    cout << "Using only: " << updatesPerEpoch*patternsPerUpdate << 
      " patterns on available: " << NPatterns << endl;
  } else if (updatesPerEpoch*patternsPerUpdate>NPatterns) 
  {
    cout << " Trying to use " << updatesPerEpoch*patternsPerUpdate << 
      " patterns, but available: " << NPatterns << endl;
    return -100;
  }
  
  for( Int_t iPattern = 0; iPattern < updatesPerEpoch*patternsPerUpdate;
       iPattern++ )
  {
    for( Int_t i = 0; i < inputDim; i++ )
    {
      JNDAT1.OIN[ i ] = float ( GetInputTrainSet( iPattern, i ) );
    }
    
    NWJNWGT.OWGT = GetEventWeightTrainSet( iPattern );
    
    for( Int_t j = 0; j < outputDim; j++ )
    {
      JNDAT1.OUT[ j ] = float ( GetOutputTrainSet( iPattern, j ) );
    }
 
    JNTRAL();
  }
 
  return GetPARJN(8);
}

UnlockInit()

void TJetNet::UnlockInit ( void )

inline

Definition at line 138 of file TJetNet.h.

{ m_InitLocked = kFALSE; };

writeNetworkInfo()

void TJetNet::writeNetworkInfo

(

Int_t

typeOfInfo = 0

)

Definition at line 664 of file TJetNet.cxx.

{
  cout << " Invoking info of type: " << typeOfInfo << endl;
  JNSTAT(typeOfInfo);
}  

Member Data Documentation

m_CurrentEpoch

Int_t TJetNet::m_CurrentEpoch

private

Definition at line 186 of file TJetNet.h.

m_Debug

Bool_t TJetNet::m_Debug

private

Definition at line 187 of file TJetNet.h.

m_enActFunction

TActivationFunction TJetNet::m_enActFunction {}

private

Definition at line 170 of file TJetNet.h.

{};

m_Epochs

Int_t TJetNet::m_Epochs

private

Definition at line 185 of file TJetNet.h.

m_HiddenLayerDim

Int_t TJetNet::m_HiddenLayerDim

private

Definition at line 183 of file TJetNet.h.

m_InitLocked

Bool_t TJetNet::m_InitLocked

private

Definition at line 189 of file TJetNet.h.

m_InputDim

Int_t TJetNet::m_InputDim

private

Definition at line 182 of file TJetNet.h.

m_IsInitialized

Bool_t TJetNet::m_IsInitialized

private

Definition at line 188 of file TJetNet.h.

m_LayerCount

Int_t TJetNet::m_LayerCount

private

Definition at line 172 of file TJetNet.h.

m_NormalizeOutput

Bool_t TJetNet::m_NormalizeOutput

private

Definition at line 190 of file TJetNet.h.

m_OutputDim

Int_t TJetNet::m_OutputDim

private

Definition at line 184 of file TJetNet.h.

m_pInputTestSet

TNeuralDataSet* TJetNet::m_pInputTestSet

private

Definition at line 177 of file TJetNet.h.

m_pInputTrainSet

TNeuralDataSet* TJetNet::m_pInputTrainSet

private

Array which contains the number of units in each layer.

Definition at line 175 of file TJetNet.h.

m_pLayers

Int_t* TJetNet::m_pLayers

private

Definition at line 173 of file TJetNet.h.

m_pOutputTestSet

TNeuralDataSet* TJetNet::m_pOutputTestSet

private

Definition at line 178 of file TJetNet.h.

m_pOutputTrainSet

TNeuralDataSet* TJetNet::m_pOutputTrainSet

private

Definition at line 176 of file TJetNet.h.

m_TestSetCnt

Int_t TJetNet::m_TestSetCnt

private

Definition at line 180 of file TJetNet.h.

m_TrainSetCnt

Int_t TJetNet::m_TrainSetCnt

private

Definition at line 180 of file TJetNet.h.

---

The documentation for this class was generated from the following files:

• TJetNet.h
• TJetNet.cxx