mitkGeneralizedLinearModel.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 mitkGeneralizedLinearModel_h
#define mitkGeneralizedLinearModel_h
 
#include <MitkCLImportanceWeightingExports.h>
 
#include <vnl/vnl_matrix.h>
#include <vnl/vnl_vector.h>
 
namespace mitk
{
  class MITKCLIMPORTANCEWEIGHTING_EXPORT GeneralizedLinearModel
  {
  public:
    GeneralizedLinearModel (const vnl_matrix<double> &xData, const vnl_vector<double> &yData, bool addConstantColumn=true);
 
    double Predict(const vnl_vector<double> &c);
 
    vnl_vector<double> Predict(const vnl_matrix<double> &x);
 
    vnl_vector<double> ExpMu(const vnl_matrix<double> &x);
 
    vnl_vector<double> B();
 
  private:
 
    // Estimates the rank of the matrix and creates a permutation vector so
    // that the most important columns are first. Depends on a QR-algorithm.
    void EstimatePermutation(const vnl_matrix<double> &xData);
 
    vnl_vector<unsigned int>    m_Permutation;        // Holds a permutation matrix which is used during calculation of B
    vnl_vector<double>          m_B;                  // B-Values. Linear componentn of the model.
    bool                        m_AddConstantColumn;  // If true, a constant value is added to each row
//    int                         m_Rank;               // The estimated input rank of the matrix.
  };
}
 
#endif