Example #1
0
void RandomForestRegressor::build_tree(suml::basic::Tree<float> * & tree,
		std::vector<std::vector<float> > &feature,
		std::vector<float> &label) {

	tree = new suml::tree::RegressionTree(get_tree_depth(),get_tree_node_cnt(), get_multi_thread_on(), true);
	tree->getMinSampleCnt() = get_min_sample_cnt();
	tree->setData(feature, label);
	tree->train();
}
Example #2
0
void print_tree(tree *pTree)
{
	get_tree_depth(pTree);
	int temp = DEPTH;
	//printf("node num:%d\n", pTree->nodeNum);
	while(temp-- > 0)
		{
			printf("Row %d", mylow);
			traverse(pTree);
			printf("\n");
			mylow++;
			reserve_prev_node_col = 1;
		}

	mylow = 1;
	//reserve_prev_node_col = 1;
}