/usr/include/shogun/multiclass/MulticlassOCAS.h is in libshogun-dev 3.2.0-7.3build4.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 | /*
* 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/common.h>
#include <shogun/features/DotFeatures.h>
#include <shogun/lib/external/libocas.h>
#include <shogun/machine/LinearMulticlassMachine.h>
namespace shogun
{
/** @brief multiclass OCAS wrapper */
class CMulticlassOCAS : public CLinearMulticlassMachine
{
public:
MACHINE_PROBLEM_TYPE(PT_MULTICLASS)
/** default constructor */
CMulticlassOCAS();
/** standard constructor
* @param C C regularication constant value
* @param features features
* @param labs labels
*/
CMulticlassOCAS(float64_t C, CDotFeatures* features, CLabels* labs);
/** destructor */
virtual ~CMulticlassOCAS();
/** get name */
virtual const char* get_name() const
{
return "MulticlassOCAS";
}
/** set C
* @param C C value
*/
inline void set_C(float64_t C)
{
ASSERT(C>0)
m_C = C;
}
/** get C
* @return C value
*/
inline float64_t get_C() const { return m_C; }
/** set epsilon
* @param epsilon epsilon value
*/
inline void set_epsilon(float64_t epsilon)
{
ASSERT(epsilon>0)
m_epsilon = epsilon;
}
/** get epsilon
* @return epsilon value
*/
inline float64_t get_epsilon() const { return m_epsilon; }
/** set max iter
* @param max_iter max iter value
*/
inline void set_max_iter(int32_t max_iter)
{
ASSERT(max_iter>0)
m_max_iter = max_iter;
}
/** get max iter
* @return max iter value
*/
inline int32_t get_max_iter() const { return m_max_iter; }
/** set method
* @param method method value
*/
inline void set_method(int32_t method)
{
ASSERT(method==0 || method==1)
m_method = method;
}
/** get method
* @return method value
*/
inline int32_t get_method() const { return m_method; }
/** set buf size
* @param buf_size buf size value
*/
inline void set_buf_size(int32_t buf_size)
{
ASSERT(buf_size>0)
m_buf_size = buf_size;
}
/** get buf size
* @return buf_size value
*/
inline int32_t get_buf_size() const { return m_buf_size; }
protected:
/** train machine */
virtual bool train_machine(CFeatures* data = NULL);
/** update W */
static float64_t msvm_update_W(float64_t t, void* user_data);
/** full compute W */
static void msvm_full_compute_W(float64_t *sq_norm_W, float64_t *dp_WoldW,
float64_t *alpha, uint32_t nSel, void* user_data);
/** full add new cut */
static int msvm_full_add_new_cut(float64_t *new_col_H, uint32_t *new_cut,
uint32_t nSel, void* user_data);
/** full compute output */
static int msvm_full_compute_output(float64_t *output, void* user_data);
/** sort */
static int msvm_sort_data(float64_t* vals, float64_t* data, uint32_t size);
/** print nothing */
static void msvm_print(ocas_return_value_T value);
private:
/** register parameters */
void register_parameters();
protected:
/** regularization constant for each machine */
float64_t m_C;
/** tolerance */
float64_t m_epsilon;
/** max number of iterations */
int32_t m_max_iter;
/** method */
int32_t m_method;
/** buffer size */
int32_t m_buf_size;
};
}
#endif
|