SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
classifier
vw
VwRegressor.h
浏览该文件的文档.
1
/*
2
* Copyright (c) 2009 Yahoo! Inc. All rights reserved. The copyrights
3
* embodied in the content of this file are licensed under the BSD
4
* (revised) open source license.
5
*
6
* This program is free software; you can redistribute it and/or modify
7
* it under the terms of the GNU General Public License as published by
8
* the Free Software Foundation; either version 3 of the License, or
9
* (at your option) any later version.
10
*
11
* Written (W) 2011 Shashwat Lal Das
12
* Adaptation of Vowpal Wabbit v5.1.
13
* Copyright (C) 2011 Berlin Institute of Technology and Max-Planck-Society.
14
*/
15
16
#ifndef _VW_REGRESSOR_H__
17
#define _VW_REGRESSOR_H__
18
19
#include <
shogun/base/SGObject.h
>
20
#include <
shogun/lib/DataType.h
>
21
#include <
shogun/classifier/vw/VwEnvironment.h
>
22
#include <
shogun/loss/LossFunction.h
>
23
24
namespace
shogun
25
{
26
35
class
CVwRegressor
:
public
CSGObject
36
{
37
public
:
41
CVwRegressor
();
42
48
CVwRegressor
(
CVwEnvironment
* env_to_use);
49
53
virtual
~CVwRegressor
();
54
63
inline
float64_t
get_loss
(
float64_t
prediction,
float64_t
label)
64
{
65
return
loss
->
loss
(prediction, label);
66
}
67
78
inline
float64_t
get_update
(
float64_t
prediction,
float64_t
label,
79
float64_t
eta_t,
float64_t
norm
)
80
{
81
return
loss
->
get_update
(prediction, label, eta_t, norm);
82
}
83
90
virtual
void
dump_regressor
(
char
* reg_name,
bool
as_text);
91
97
virtual
void
load_regressor
(
char
* file_name);
98
103
virtual
const
char
*
get_name
()
const
{
return
"VwRegressor"
; }
104
110
virtual
void
init(
CVwEnvironment
* env_to_use = NULL);
111
112
public
:
114
float32_t
**
weight_vectors
;
116
CLossFunction
*
loss
;
117
118
protected
:
120
CVwEnvironment
*
env
;
121
};
122
123
}
124
#endif // _VW_REGRESSOR_H__
SHOGUN
机器学习工具包 - 项目文档