Commit 22142246 authored by Maria Dimashova's avatar Maria Dimashova

fixed priors usage in ertrees

parent d8417af0
...@@ -954,7 +954,7 @@ CvDTreeSplit* CvForestERTree::find_split_ord_class( CvDTreeNode* node, int vi, f ...@@ -954,7 +954,7 @@ CvDTreeSplit* CvForestERTree::find_split_ord_class( CvDTreeNode* node, int vi, f
int r = responses[si]; int r = responses[si];
float val = values[si]; float val = values[si];
int m = missing[si]; int m = missing[si];
double p = priors[si]; double p = priors[r];
if (m) continue; if (m) continue;
if ( val < split_val ) if ( val < split_val )
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment