Merge pull request #16251 from pwuertz:rtrees_set_rng

pull/16263/head
Alexander Alekhin 5 years ago
commit b4759d7272
  1. 4
      modules/ml/src/rtrees.cpp

@ -90,12 +90,12 @@ public:
CV_TRACE_FUNCTION();
DTreesImpl::clear();
oobError = 0.;
rng = RNG((uint64)-1);
}
const vector<int>& getActiveVars() CV_OVERRIDE
{
CV_TRACE_FUNCTION();
RNG &rng = theRNG();
int i, nvars = (int)allVars.size(), m = (int)activeVars.size();
for( i = 0; i < nvars; i++ )
{
@ -134,6 +134,7 @@ public:
bool train( const Ptr<TrainData>& trainData, int flags ) CV_OVERRIDE
{
CV_TRACE_FUNCTION();
RNG &rng = theRNG();
CV_Assert(!trainData.empty());
startTraining(trainData, flags);
int treeidx, ntrees = (rparams.termCrit.type & TermCriteria::COUNT) != 0 ?
@ -424,7 +425,6 @@ public:
double oobError;
vector<float> varImportance;
vector<int> allVars, activeVars;
RNG rng;
};

Loading…
Cancel
Save