SHOGUN
3.2.1
首页
相关页面
模块
类
文件
文件列表
文件成员
全部
类
命名空间
文件
函数
变量
类型定义
枚举
枚举值
友元
宏定义
组
页
src
shogun
transfer
multitask
MultitaskClusteredLogisticRegression.h
浏览该文件的文档.
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
* Copyright (C) 2012 Sergey Lisitsyn
8
*/
9
10
#ifndef MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
11
#define MULTITASKCLUSTEREDLOGISTICREGRESSION_H_
12
13
#include <
shogun/transfer/multitask/MultitaskLogisticRegression.h
>
14
15
namespace
shogun
16
{
23
class
CMultitaskClusteredLogisticRegression
:
public
CMultitaskLogisticRegression
24
{
25
26
public
:
27
MACHINE_PROBLEM_TYPE
(
PT_BINARY
)
28
29
30
CMultitaskClusteredLogisticRegression
();
31
41
CMultitaskClusteredLogisticRegression
(
42
float64_t
rho1,
float64_t
rho2,
CDotFeatures
* training_data,
43
CBinaryLabels
* training_labels,
CTaskGroup
* task_group,
44
int32_t num_clusters);
45
47
virtual
~CMultitaskClusteredLogisticRegression
();
48
53
int32_t
get_rho1
()
const
;
54
58
void
set_rho1
(
float64_t
rho1);
59
62
int32_t
get_rho2
()
const
;
63
67
void
set_rho2
(
float64_t
rho2);
68
73
int32_t
get_num_clusters
()
const
;
74
78
void
set_num_clusters
(int32_t num_clusters);
79
84
virtual
const
char
*
get_name
()
const
85
{
86
return
"MultitaskClusteredLogisticRegression"
;
87
}
88
89
protected
:
90
95
virtual
bool
train_machine
(
CFeatures
* data=NULL);
96
101
virtual
bool
train_locked_implementation
(
SGVector<index_t>
* tasks);
102
103
protected
:
104
106
float64_t
m_rho1
;
107
109
float64_t
m_rho2
;
110
112
int32_t
m_num_clusters
;
113
};
114
}
115
#endif
SHOGUN
机器学习工具包 - 项目文档