LDA.h

Go to the documentation of this file.

/*
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * Written (W) 1999-2009 Soeren Sonnenburg
 * Copyright (C) 1999-2009 Fraunhofer Institute FIRST and Max-Planck-Society
 */

#ifndef _LDA_H___
#define _LDA_H___

#include "lib/config.h"

#ifdef HAVE_LAPACK
#include "lib/common.h"
#include "features/Features.h"
#include "features/SimpleFeatures.h"
#include "classifier/LinearClassifier.h"

namespace shogun
{
    template <class ST> class CSimpleFeatures;
#define IGNORE_IN_CLASSLIST
IGNORE_IN_CLASSLIST class CLDA : public CLinearClassifier
{
    public:
        CLDA(float64_t gamma=0);

        CLDA(float64_t gamma, CSimpleFeatures<float64_t>* traindat, CLabels* trainlab);
        virtual ~CLDA();

        inline void set_gamma(float64_t gamma)
        {
            m_gamma=gamma;
        }

        inline float64_t get_gamma()
        {
            return m_gamma;
        }

        virtual bool train(CFeatures* data=NULL);

        inline virtual EClassifierType get_classifier_type()
        {
            return CT_LDA;
        }

        virtual inline void set_features(CDotFeatures* feat)
        {
            if (feat->get_feature_class() != C_SIMPLE ||
                feat->get_feature_type() != F_DREAL)
                SG_ERROR("LDA requires SIMPLE REAL valued features\n");

            CLinearClassifier::set_features(feat);
        }

        inline virtual const char* get_name() const { return "LDA"; }

    protected:
        float64_t m_gamma;
};
}
#endif
#endif