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


#ifndef _MULTICLASSOCAS_H___
#define _MULTICLASSOCAS_H___

#include <shogun/lib/config.h>
#ifdef USE_GPL_SHOGUN

#include <shogun/lib/common.h>
#include <shogun/features/DotFeatures.h>
#include <shogun/lib/external/libocas.h>
#include <shogun/machine/LinearMulticlassMachine.h>

namespace shogun
{

class CMulticlassOCAS : public CLinearMulticlassMachine
{
    public:
        MACHINE_PROBLEM_TYPE(PT_MULTICLASS)

        
        CMulticlassOCAS();

        CMulticlassOCAS(float64_t C, CDotFeatures* features, CLabels* labs);

        virtual ~CMulticlassOCAS();

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

        inline void set_C(float64_t C)
        {
            ASSERT(C>0)
            m_C = C;
        }
        inline float64_t get_C() const { return m_C; }

        inline void set_epsilon(float64_t epsilon)
        {
            ASSERT(epsilon>0)
            m_epsilon = epsilon;
        }
        inline float64_t get_epsilon() const { return m_epsilon; }

        inline void set_max_iter(int32_t max_iter)
        {
            ASSERT(max_iter>0)
            m_max_iter = max_iter;
        }
        inline int32_t get_max_iter() const { return m_max_iter; }

        inline void set_method(int32_t method)
        {
            ASSERT(method==0 || method==1)
            m_method = method;
        }
        inline int32_t get_method() const { return m_method; }

        inline void set_buf_size(int32_t buf_size)
        {
            ASSERT(buf_size>0)
            m_buf_size = buf_size;
        }
        inline int32_t get_buf_size() const { return m_buf_size; }

protected:

        virtual bool train_machine(CFeatures* data = NULL);

        static float64_t msvm_update_W(float64_t t, void* user_data);

        static void msvm_full_compute_W(float64_t *sq_norm_W, float64_t *dp_WoldW,
                                        float64_t *alpha, uint32_t nSel, void* user_data);

        static int msvm_full_add_new_cut(float64_t *new_col_H, uint32_t *new_cut,
                                         uint32_t nSel, void* user_data);

        static int msvm_full_compute_output(float64_t *output, void* user_data);

        static int msvm_sort_data(float64_t* vals, float64_t* data, uint32_t size);

        static void msvm_print(ocas_return_value_T value);

private:

        void register_parameters();

protected:

        float64_t m_C;

        float64_t m_epsilon;

        int32_t m_max_iter;

        int32_t m_method;

        int32_t m_buf_size;
};
}
#endif

#endif