Commits

Aleš Erjavec  committed 873ff9b

Set distribution variable in 'SimpleTreeLearner.classDistribution'.

  • Participants
  • Parent commits d65c874

Comments (0)

Files changed (1)

File source/orange/tdidt_simple.cpp

 		dist = predict_classification(ex, tree, &free_dist, this->cls_vals);
 
 		PDistribution pdist = mlnew TDiscDistribution(this->cls_vals, 0.0);
+		pdist->variable = this->classVar;
 		for (i = 0; i < this->cls_vals; i++)
 			pdist->setint(i, dist[i]);
 		pdist->normalize();