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 <shogun/lib/config.h>

#ifdef HAVE_LAPACK
#include <shogun/lib/common.h>
#include <shogun/features/Features.h>
#include <shogun/features/SimpleFeatures.h>
#include <shogun/machine/LinearMachine.h>

namespace shogun
{
    template <class ST> class CSimpleFeatures;
class CLDA : public CLinearMachine
{
    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;
        }

        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");

            CLinearMachine::set_features(feat);
        }

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

    protected:
        virtual bool train_machine(CFeatures* data=NULL);

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