MultiClassSVM.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 _MULTICLASSSVM_H___
#define _MULTICLASSSVM_H___

#include <shogun/lib/common.h>
#include <shogun/features/Features.h>
#include <shogun/classifier/svm/SVM.h>

namespace shogun
{

#ifndef DOXYGEN_SHOULD_SKIP_THIS
enum EMultiClassSVM
{
    ONE_VS_REST,
    ONE_VS_ONE,
};
#endif

class CSVM;

class CMultiClassSVM : public CSVM
{
    public:
        CMultiClassSVM();

        CMultiClassSVM(EMultiClassSVM type);

        CMultiClassSVM(
            EMultiClassSVM type, float64_t C, CKernel* k, CLabels* lab);
        virtual ~CMultiClassSVM();

        bool create_multiclass_svm(int32_t num_classes);

        bool set_svm(int32_t num, CSVM* svm);

        CSVM* get_svm(int32_t num)
        {
            ASSERT(m_svms && m_num_svms>0);
            ASSERT(num>=0 && num<m_num_svms);
            SG_REF(m_svms[num]);
            return m_svms[num];
        }

        int32_t inline get_num_svms()
        {
            return m_num_svms;
        }

        void cleanup();

        virtual CLabels* apply();

        virtual CLabels* apply(CFeatures* data)
        {
            return CKernelMachine::apply(data);
        }

        virtual float64_t apply(int32_t num);

        virtual CLabels* classify_one_vs_rest();

        virtual float64_t classify_example_one_vs_rest(int32_t num);

        CLabels* classify_one_vs_one();

        float64_t classify_example_one_vs_one(int32_t num);

        bool load(FILE* svm_file);

        bool save(FILE* svm_file);

        inline EMultiClassSVM get_multiclass_type() { return multiclass_type; }

    private:
        void init();

    protected:
        EMultiClassSVM multiclass_type;

        int32_t m_num_classes;
        int32_t m_num_svms;
        CSVM** m_svms;
};
}
#endif