Qualia
0.2
Main Page
Related Pages
Namespaces
Classes
Files
File List
File Members
src
qualia
computer
SubDataSet.h
Go to the documentation of this file.
1
/*
2
* SubDataSet.h
3
*
4
* (c) 2013 Sofian Audry -- info(@)sofianaudry(.)com
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
* This program is distributed in the hope that it will be useful,
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
* GNU General Public License for more details.
15
*
16
* You should have received a copy of the GNU General Public License
17
* along with this program. If not, see <http://www.gnu.org/licenses/>.
18
*/
19
20
#ifndef SUBDATASET_H_
21
#define SUBDATASET_H_
22
23
#include <
qualia/learning/DataSet.h
>
24
#include <
qualia/computer/random_shuffle.h
>
25
26
class
SubDataSet
:
public
DataSet
{
27
public
:
28
DataSet
*
_dataSet
;
29
int
*
_indices
;
30
31
SubDataSet
(
DataSet
* dataSet,
int
* indices,
int
nExamples
);
32
virtual
~SubDataSet
();
33
34
// To be called before training. Should allocate example, among other things.
35
virtual
void
init
();
36
37
// To be called before every episode.
38
virtual
void
reset
();
39
40
// Sets this->example to given index t.
41
virtual
void
setExample
(
int
t);
42
43
static
SubDataSet
*
getShuffledDataSet
(
DataSet
* dataSet,
int
nExamples
=-1);
44
45
static
void
getStandardDataSets
(
SubDataSet
** train,
SubDataSet
** valid,
SubDataSet
** test,
46
DataSet
* dataSet,
47
float
propTrain=0.5f,
float
propValid=0.2f,
float
propTest=0.3f);
48
};
49
50
#endif
/* SUBDATASET_H_ */
Generated on Fri Oct 11 2013 19:01:21 for Qualia by
1.8.3.1