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 //mitkImpurityLoss_h