SHOGUN
v2.0.0
Main Page
Related Pages
Modules
Classes
Files
File List
File Members
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Friends
Macros
Groups
Pages
src
shogun
multiclass
ecoc
ECOCStrategy.h
Go to the documentation of this file.
1
/*
2
* This program is free software; you can redistribute it and/or modify
3
* it under the terms of the GNU General Public License as published by
4
* the Free Software Foundation; either version 3 of the License, or
5
* (at your option) any later version.
6
*
7
* Written (W) 2012 Chiyuan Zhang
8
* Copyright (C) 2012 Chiyuan Zhang
9
*/
10
11
#ifndef ECOCSTRATEGY_H__
12
#define ECOCSTRATEGY_H__
13
14
#include <
shogun/multiclass/MulticlassStrategy.h
>
15
#include <
shogun/multiclass/ecoc/ECOCEncoder.h
>
16
#include <
shogun/multiclass/ecoc/ECOCDecoder.h
>
17
18
namespace
shogun
19
{
20
22
class
CECOCStrategy
:
public
CMulticlassStrategy
23
{
24
public
:
26
CECOCStrategy
();
27
29
CECOCStrategy
(
CECOCEncoder
*encoder,
CECOCDecoder
*decoder);
30
32
virtual
~CECOCStrategy
();
33
35
virtual
const
char
*
get_name
()
const
36
{
37
return
"ECOCStrategy"
;
38
}
39
41
virtual
void
train_start
(
CMulticlassLabels
*orig_labels,
CBinaryLabels
*train_labels);
42
44
virtual
bool
train_has_more
();
45
49
virtual
SGVector<int32_t>
train_prepare_next
();
50
54
virtual
int32_t
decide_label
(
SGVector<float64_t>
outputs);
55
58
virtual
int32_t
get_num_machines
();
59
60
protected
:
62
CECOCEncoder
*
m_encoder
;
64
CECOCDecoder
*
m_decoder
;
65
67
SGMatrix<int32_t>
m_codebook
;
68
69
private
:
71
void
init();
72
};
73
74
75
}
76
77
#endif
/* end of include guard: ECOCSTRATEGY_H__ */
SHOGUN
Machine Learning Toolbox - Documentation