ConjugateIndex.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) 2011 Sergey Lisitsyn
 * Copyright (C) 2011 Sergey Lisitsyn
 */

#ifndef CONJUGATEINDEX_H_
#define CONJUGATEINDEX_H_
#ifdef HAVE_LAPACK
#include <shogun/machine/Machine.h>
#include <shogun/mathematics/Math.h>
#include <shogun/features/DenseFeatures.h>

namespace shogun
{

class CLabels;
class CDotFeatures;
class CFeatures;

class CConjugateIndex : public CMachine
{

public:
    MACHINE_PROBLEM_TYPE(PT_MULTICLASS)

    
    CConjugateIndex();

    CConjugateIndex(CFeatures* train_features, CLabels* train_labels);

    virtual ~CConjugateIndex();

    virtual void set_features(CFeatures* features);

    virtual CDenseFeatures<float64_t>* get_features();

    virtual CMulticlassLabels* apply_multiclass(CFeatures* data=NULL);

    virtual float64_t apply_one(int32_t idx);

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

    virtual EMachineType get_classifier_type() { return CT_CONJUGATEINDEX; };

protected:

    virtual bool train_machine(CFeatures* data=NULL);

    void clean_classes();

    float64_t conjugate_index(SGVector<float64_t> feature_vector, int32_t label);

protected:

    int32_t m_num_classes;

    SGVector<float64_t> m_feature_vector;

    SGMatrix<float64_t>* m_classes;

    CDenseFeatures<float64_t>* m_features;

};

}
#endif /* HAVE_LAPACK */
#endif /* CONJUGATEINDEX_H_ */