TMLPAnalyzer.h

Go to the documentation of this file.

// @(#)root/mlp:$Name$:$Id$
// Author: Christophe.Delaere@cern.ch   25/04/04

/*************************************************************************
 * Copyright (C) 1995-2003, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

#ifndef ROOT_TObject
#include "TObject.h"
#endif

class TTree;
class TNeuron;
class TSynapse;
class TMultiLayerPerceptron;
class TProfile;
class THStack;

//____________________________________________________________________
//
// TMLPAnalyzer
//
// This utility class contains a set of tests usefull when developing
// a neural network.
// It allows you to check for unneeded variables, and to control
// the network structure.
//
//--------------------------------------------------------------------

class TMLPAnalyzer: public TObject
{

private:
    TMultiLayerPerceptron *fNetwork;
    TTree *fAnalysisTree;
    TTree *fIOTree;

protected:
    Int_t GetLayers();
    Int_t GetNeurons(Int_t layer);
    TString GetNeuronFormula(Int_t idx);
    const char* GetInputNeuronTitle(Int_t in);
    const char* GetOutputNeuronTitle(Int_t out);

public:
    TMLPAnalyzer(TMultiLayerPerceptron& net) :
            fNetwork(&net), fAnalysisTree(0), fIOTree(0)
    {
    }
    TMLPAnalyzer(TMultiLayerPerceptron* net) :
            fNetwork(net), fAnalysisTree(0), fIOTree(0)
    {
    }
    virtual ~TMLPAnalyzer();
    void DrawNetwork(Int_t neuron, const char* signal, const char* bg);
    void DrawDInput(Int_t i);
    void DrawDInputs();
    TProfile* DrawTruthDeviation(Int_t outnode = 0, Option_t *option = "");
    THStack* DrawTruthDeviations(Option_t *option = "");
    TProfile* DrawTruthDeviationInOut(Int_t innode, Int_t outnode = 0,
            Option_t *option = "");
    THStack* DrawTruthDeviationInsOut(Int_t outnode = 0, Option_t *option = "");

    void CheckNetwork();
    void GatherInformations();
    TTree* GetIOTree() const
    {
        return fIOTree;
    }

    ClassDef(TMLPAnalyzer, 0)
};