Merge pull request #26046 from ivafanas:speedup_random_forest_get_votes

speedup random forest getVotes method.
pull/26051/head
Alexander Smorkalov 6 months ago committed by GitHub
commit ec2e2da1fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 6
      modules/ml/src/rtrees.cpp

@ -388,9 +388,10 @@ public:
results = output.getMat();
for( i = 0; i < nsamples; i++ )
{
const Mat sampleRow = samples.row(i);
for( j = 0; j < ntrees; j++ )
{
float val = predictTrees( Range(j, j+1), samples.row(i), flags);
float val = predictTrees( Range(j, j+1), sampleRow, flags);
results.at<float> (i, j) = val;
}
}
@ -408,9 +409,10 @@ public:
for( i = 0; i < nsamples; i++ )
{
votes.clear();
const Mat sampleRow = samples.row(i);
for( j = 0; j < ntrees; j++ )
{
int val = (int)predictTrees( Range(j, j+1), samples.row(i), flags);
int val = (int)predictTrees( Range(j, j+1), sampleRow, flags);
votes.push_back(val);
}

Loading…
Cancel
Save