SHOGUN
v3.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
tree
RelaxedTreeUtil.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 RELAXEDTREEUTIL_H__
12
#define RELAXEDTREEUTIL_H__
13
14
#include <
shogun/machine/BaseMulticlassMachine.h
>
15
#include <
shogun/features/DenseFeatures.h
>
16
#include <
shogun/labels/MulticlassLabels.h
>
17
#include <
shogun/lib/SGMatrix.h
>
18
19
namespace
shogun
20
{
21
23
class
RelaxedTreeUtil
24
{
25
public
:
26
34
SGMatrix<float64_t>
estimate_confusion_matrix
(
CBaseMulticlassMachine
*machine,
CFeatures
*X,
CMulticlassLabels
*Y, int32_t num_classes);
35
42
void
get_confusion_matrix
(
SGMatrix<float64_t>
&conf_mat,
CMulticlassLabels
*gt,
CMulticlassLabels
*pred);
43
};
44
45
}
/* shogun */
46
47
#endif
/* end of include guard: RELAXEDTREEUTIL_H__ */
48
SHOGUN
Machine Learning Toolbox - Documentation