Lines Matching refs:fs
1290 void CvDTreeTrainData::write_params( CvFileStorage* fs ) const
1298 cvWriteInt( fs, "is_classifier", is_classifier ? 1 : 0 );
1299 cvWriteInt( fs, "var_all", var_all );
1300 cvWriteInt( fs, "var_count", var_count );
1301 cvWriteInt( fs, "ord_var_count", ord_var_count );
1302 cvWriteInt( fs, "cat_var_count", cat_var_count );
1304 cvStartWriteStruct( fs, "training_params", CV_NODE_MAP );
1305 cvWriteInt( fs, "use_surrogates", params.use_surrogates ? 1 : 0 );
1309 cvWriteInt( fs, "max_categories", params.max_categories );
1313 cvWriteReal( fs, "regression_accuracy", params.regression_accuracy );
1316 cvWriteInt( fs, "max_depth", params.max_depth );
1317 cvWriteInt( fs, "min_sample_count", params.min_sample_count );
1318 cvWriteInt( fs, "cross_validation_folds", params.cv_folds );
1322 cvWriteInt( fs, "use_1se_rule", params.use_1se_rule ? 1 : 0 );
1323 cvWriteInt( fs, "truncate_pruned_tree", params.truncate_pruned_tree ? 1 : 0 );
1327 cvWrite( fs, "priors", priors );
1329 cvEndWriteStruct( fs );
1332 cvWrite( fs, "var_idx", var_idx );
1334 cvStartWriteStruct( fs, "var_type", CV_NODE_SEQ+CV_NODE_FLOW );
1337 cvWriteInt( fs, 0, var_type->data.i[vi] >= 0 );
1339 cvEndWriteStruct( fs );
1344 cvWrite( fs, "cat_count", cat_count );
1345 cvWrite( fs, "cat_map", cat_map );
1352 void CvDTreeTrainData::read_params( CvFileStorage* fs, CvFileNode* node )
1362 is_classifier = (cvReadIntByName( fs, node, "is_classifier" ) != 0);
1363 var_all = cvReadIntByName( fs, node, "var_all" );
1364 var_count = cvReadIntByName( fs, node, "var_count", var_all );
1365 cat_var_count = cvReadIntByName( fs, node, "cat_var_count" );
1366 ord_var_count = cvReadIntByName( fs, node, "ord_var_count" );
1368 tparams_node = cvGetFileNodeByName( fs, node, "training_params" );
1372 params.use_surrogates = cvReadIntByName( fs, tparams_node, "use_surrogates", 1 ) != 0;
1376 params.max_categories = cvReadIntByName( fs, tparams_node, "max_categories" );
1381 (float)cvReadRealByName( fs, tparams_node, "regression_accuracy" );
1384 params.max_depth = cvReadIntByName( fs, tparams_node, "max_depth" );
1385 params.min_sample_count = cvReadIntByName( fs, tparams_node, "min_sample_count" );
1386 params.cv_folds = cvReadIntByName( fs, tparams_node, "cross_validation_folds" );
1390 params.use_1se_rule = cvReadIntByName( fs, tparams_node, "use_1se_rule" ) != 0;
1392 cvReadIntByName( fs, tparams_node, "truncate_pruned_tree" ) != 0;
1395 priors = (CvMat*)cvReadByName( fs, tparams_node, "priors" );
1404 CV_CALL( var_idx = (CvMat*)cvReadByName( fs, node, "var_idx" ));
1424 vartype_node = cvGetFileNodeByName( fs, node, "var_type" );
1453 CV_CALL( cat_count = (CvMat*)cvReadByName( fs, node, "cat_count" ));
1454 CV_CALL( cat_map = (CvMat*)cvReadByName( fs, node, "cat_map" ));
3785 void CvDTree::write_split( CvFileStorage* fs, CvDTreeSplit* split ) const
3789 cvStartWriteStruct( fs, 0, CV_NODE_MAP + CV_NODE_FLOW );
3790 cvWriteInt( fs, "var", split->var_idx );
3791 cvWriteReal( fs, "quality", split->quality );
3804 cvStartWriteStruct( fs, default_dir*(split->inversed ? -1 : 1) > 0 ?
3811 cvWriteInt( fs, 0, i );
3813 cvEndWriteStruct( fs );
3816 cvWriteReal( fs, !split->inversed ? "le" : "gt", split->ord.c );
3818 cvEndWriteStruct( fs );
3822 void CvDTree::write_node( CvFileStorage* fs, CvDTreeNode* node ) const
3826 cvStartWriteStruct( fs, 0, CV_NODE_MAP );
3828 cvWriteInt( fs, "depth", node->depth );
3829 cvWriteInt( fs, "sample_count", node->sample_count );
3830 cvWriteReal( fs, "value", node->value );
3833 cvWriteInt( fs, "norm_class_idx", node->class_idx );
3835 cvWriteInt( fs, "Tn", node->Tn );
3836 cvWriteInt( fs, "complexity", node->complexity );
3837 cvWriteReal( fs, "alpha", node->alpha );
3838 cvWriteReal( fs, "node_risk", node->node_risk );
3839 cvWriteReal( fs, "tree_risk", node->tree_risk );
3840 cvWriteReal( fs, "tree_error", node->tree_error );
3844 cvStartWriteStruct( fs, "splits", CV_NODE_SEQ );
3847 write_split( fs, split );
3849 cvEndWriteStruct( fs );
3852 cvEndWriteStruct( fs );
3856 void CvDTree::write_tree_nodes( CvFileStorage* fs ) const
3870 write_node( fs, node );
3890 void CvDTree::write( CvFileStorage* fs, const char* name ) const
3896 cvStartWriteStruct( fs, name, CV_NODE_MAP, CV_TYPE_NAME_ML_TREE );
3899 data->write_params( fs );
3901 //cvWrite( fs, "var_importance", var_importance );
3902 write( fs );
3904 cvEndWriteStruct( fs );
3910 void CvDTree::write( CvFileStorage* fs ) const
3916 cvWriteInt( fs, "best_tree_idx", pruned_tree_idx );
3918 cvStartWriteStruct( fs, "nodes", CV_NODE_SEQ );
3919 write_tree_nodes( fs );
3920 cvEndWriteStruct( fs );
3926 CvDTreeSplit* CvDTree::read_split( CvFileStorage* fs, CvFileNode* fnode )
3939 vi = cvReadIntByName( fs, fnode, "var", -1 );
3950 inseq = cvGetFileNodeByName( fs, fnode, "in" );
3953 inseq = cvGetFileNodeByName( fs, fnode, "not_in" );
3996 cmp_node = cvGetFileNodeByName( fs, fnode, "le" );
3999 cmp_node = cvGetFileNodeByName( fs, fnode, "gt" );
4006 split->quality = (float)cvReadRealByName( fs, fnode, "quality" );
4014 CvDTreeNode* CvDTree::read_node( CvFileStorage* fs, CvFileNode* fnode, CvDTreeNode* parent )
4029 depth = cvReadIntByName( fs, fnode, "depth", -1 );
4033 node->sample_count = cvReadIntByName( fs, fnode, "sample_count" );
4034 node->value = cvReadRealByName( fs, fnode, "value" );
4036 node->class_idx = cvReadIntByName( fs, fnode, "norm_class_idx" );
4038 node->Tn = cvReadIntByName( fs, fnode, "Tn" );
4039 node->complexity = cvReadIntByName( fs, fnode, "complexity" );
4040 node->alpha = cvReadRealByName( fs, fnode, "alpha" );
4041 node->node_risk = cvReadRealByName( fs, fnode, "node_risk" );
4042 node->tree_risk = cvReadRealByName( fs, fnode, "tree_risk" );
4043 node->tree_error = cvReadRealByName( fs, fnode, "tree_error" );
4045 splits = cvGetFileNodeByName( fs, fnode, "splits" );
4058 CV_CALL( split = read_split( fs, (CvFileNode*)reader.ptr ));
4074 void CvDTree::read_tree_nodes( CvFileStorage* fs, CvFileNode* fnode )
4092 CV_CALL( node = read_node( fs, (CvFileNode*)reader.ptr, parent != &_root ? parent : 0 ));
4114 void CvDTree::read( CvFileStorage* fs, CvFileNode* fnode )
4117 _data->read_params( fs, fnode );
4119 read( fs, fnode, _data );
4125 void CvDTree::read( CvFileStorage* fs, CvFileNode* node, CvDTreeTrainData* _data )
4136 tree_nodes = cvGetFileNodeByName( fs, node, "nodes" );
4140 pruned_tree_idx = cvReadIntByName( fs, node, "best_tree_idx", -1 );
4141 read_tree_nodes( fs, tree_nodes );
