[ VIGRA Homepage | Function Index | Class Index | Namespaces | File List | Main Page ]
#include <vigra/random_forest/rf_earlystopping.hxx>
Public Member Functions | |
template<class WeightIter , class T , class C > | |
bool | after_prediction (WeightIter, int k, MultiArrayView< 2, T, C > const &, double) |
StopAfterTree (double max_tree) |
Stop predicting after a set number of trees
StopAfterTree | ( | double | max_tree | ) |
Constructor
max_tree | number of trees to be used for prediction |
bool after_prediction | ( | WeightIter | , |
int | k, | ||
MultiArrayView< 2, T, C > const & | , | ||
double | |||
) |
called after the prediction of a tree was added to the total prediction
WeightIter | Iterator to the weights delivered by current tree. |
k | after kth tree |
prob | Total probability array |
totalCt | sum of probability array. |
Reimplemented from StopBase.
© Ullrich Köthe (ullrich.koethe@iwr.uni-heidelberg.de) |
html generated using doxygen and Python
|