iRoCS Toolbox  1.1.0
lBlitzRandomForest.hh
Go to the documentation of this file.
1 /**************************************************************************
2  *
3  * Copyright (C) 2015 Kun Liu, Thorsten Falk
4  *
5  * Image Analysis Lab, University of Freiburg, Germany
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; either version 3 of the License, or
10  * (at your option) any later version.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this program; if not, write to the Free Software Foundation,
19  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  *
21  **************************************************************************/
22 
23 #ifndef LBLITZRANDOMFOREST_H_
24 #define LBLITZRANDOMFOREST_H_
25 
26 #ifdef HAVE_CONFIG_H
27 #include <config.hh>
28 #endif
29 
30 #include <blitz/array.h>
31 #include "lRandomForest.hh"
32 
33 void trainRFSimple(
34  lRandomForest &forest, blitz::Array<float,2> &Features,
35  blitz::Array<int,1> &Labels, int m_try = -1, int labelRangeLow = 0,
36  int labelRangeHigh = 9999);
37 
38 void saveRFSimple(lRandomForest& forest, std::string fileName);
39 
40 void loadRFSimple(lRandomForest& forest, std::string fileName, int nTree);
41 
42 #endif
43 
void loadRFSimple(lRandomForest &forest, std::string fileName, int nTree)
void trainRFSimple(lRandomForest &forest, blitz::Array< float, 2 > &Features, blitz::Array< int, 1 > &Labels, int m_try=-1, int labelRangeLow=0, int labelRangeHigh=9999)
void saveRFSimple(lRandomForest &forest, std::string fileName)