mitkPUImpurityLoss.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef mitkPUImpurityLoss_h
#define mitkPUImpurityLoss_h
 
#include <vigra/multi_array.hxx>
#include <vigra/random_forest.hxx>
#include <mitkAdditionalRFData.h>
 
namespace mitk
{
 
  template<class LabelType = double>
  class PURFProblemSpec : vigra::ProblemSpec<LabelType>
  {
  public:
    vigra::ArrayVector<double>     kappa_;   // if classes have different importance
  };
 
 
    template <class TLossFunction = vigra::GiniCriterion, class TLabelContainer = vigra::MultiArrayView<2, int>, class TWeightContainer = vigra::MultiArrayView<2, double> >
    class PUImpurityLoss
    {
    public:
        typedef TLabelContainer LabelContainerType;
        typedef TWeightContainer WeightContainerType;
 
        template <class T>
        PUImpurityLoss(TLabelContainer const &labels,
                     vigra::ProblemSpec<T> const &ext,
                     AdditionalRFDataAbstract *data);
 
        void Reset();
 
        void UpdatePUCounts();
 
        template <class TDataIterator>
        double Increment(TDataIterator begin, TDataIterator end);
 
        template <class TDataIterator>
        double Decrement(TDataIterator begin, TDataIterator end);
 
        template <class TArray>
        double Init(TArray initCounts);
 
        vigra::ArrayVector<double> const& Response();
 
        void UsePointWeights(bool useWeights);
        bool IsUsingPointWeights();
 
        void SetPointWeights(TWeightContainer weight);
        WeightContainerType GetPointWeights();
 
    private:
        bool m_UsePointWeights;
        TWeightContainer m_PointWeights;
 
        //Variable of origin
        TLabelContainer const& m_Labels;
        vigra::ArrayVector<double> m_Counts;
        vigra::ArrayVector<double> m_PUCounts;
        vigra::ArrayVector<double> m_Kappa;
        vigra::ArrayVector<double> m_ClassWeights;
        double m_TotalCount;
        double m_PUTotalCount;
        int m_ClassCount;
        TLossFunction m_LossFunction;
    };
 
}
 
#include <../src/Splitter/mitkPUImpurityLoss.cpp>
 
#endif