/
AveragedPerceptron.h
84 lines (73 loc) · 2.16 KB
/
AveragedPerceptron.h
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
/*
* 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 Hidekazu Oiwa
*/
#ifndef _AVERAGEDPERCEPTRON_H___
#define _AVERAGEDPERCEPTRON_H___
#include <stdio.h>
#include <shogun/lib/common.h>
#include <shogun/features/DotFeatures.h>
#include <shogun/machine/LinearMachine.h>
namespace shogun
{
/** @brief Class Averaged Perceptron implements
* the standard linear (online) algorithm.
* Averaged perceptron is the simple extension of Perceptron.
*
* Given a maximum number of iterations (the standard averaged perceptron
* algorithm is not guaranteed to converge) and a fixed lerning rate,
* the result is a linear classifier.
*
* \sa CLinearMachine
*/
class CAveragedPerceptron : public CLinearMachine
{
public:
MACHINE_PROBLEM_TYPE(PT_BINARY);
/** default constructor */
CAveragedPerceptron();
/** constructor
*
* @param traindat training features
* @param trainlab labels for training features
*/
CAveragedPerceptron(CDotFeatures* traindat, CLabels* trainlab);
virtual ~CAveragedPerceptron();
/** get classifier type
*
* @return classifier type AVERAGEDPERCEPTRON
*/
virtual inline EMachineType get_classifier_type() { return CT_AVERAGEDPERCEPTRON; }
/** train classifier
*
* @param data training data (parameter can be avoided if distance or
* kernel-based classifiers are used and distance/kernels are
* initialized with train data)
*
* @return whether training was successful
*/
virtual bool train(CFeatures* data=NULL);
/// set learn rate of gradient descent training algorithm
inline void set_learn_rate(float64_t r)
{
learn_rate=r;
}
/// set maximum number of iterations
inline void set_max_iter(int32_t i)
{
max_iter=i;
}
/** @return object name */
inline virtual const char* get_name() const { return "AveragedPerceptron"; }
protected:
/** learning rate */
float64_t learn_rate;
/** maximum number of iterations */
int32_t max_iter;
};
}
#endif