Lines Matching refs:fs
1007 void CvGBTrees::write_params( CvFileStorage* fs ) const
1017 cvWriteString( fs, "loss_function", loss_function_type_str );
1019 cvWriteInt( fs, "loss_function", params.loss_function_type );
1021 cvWriteInt( fs, "ensemble_length", params.weak_count );
1022 cvWriteReal( fs, "shrinkage", params.shrinkage );
1023 cvWriteReal( fs, "subsample_portion", params.subsample_portion );
1024 //cvWriteInt( fs, "max_tree_depth", params.max_depth );
1025 //cvWriteString( fs, "use_surrogate_splits", params.use_surrogates ? "true" : "false");
1026 if (class_labels) cvWrite( fs, "class_labels", class_labels);
1029 data->write_params( fs );
1036 void CvGBTrees::read_params( CvFileStorage* fs, CvFileNode* fnode )
1048 CV_CALL( data->read_params(fs, fnode));
1058 temp = cvGetFileNodeByName( fs, fnode, "loss_function" );
1077 params.weak_count = cvReadIntByName( fs, fnode, "ensemble_length" );
1078 params.shrinkage = (float)cvReadRealByName( fs, fnode, "shrinkage", 0.1 );
1079 params.subsample_portion = (float)cvReadRealByName( fs, fnode, "subsample_portion", 1.0 );
1083 class_labels = (CvMat*)cvReadByName( fs, fnode, "class_labels" );
1095 void CvGBTrees::write( CvFileStorage* fs, const char* name ) const
1105 cvStartWriteStruct( fs, name, CV_NODE_MAP, CV_TYPE_NAME_ML_GBT );
1110 write_params( fs );
1111 cvWriteReal( fs, "base_value", base_value);
1112 cvWriteInt( fs, "class_count", class_count);
1117 cvStartWriteStruct( fs, s.c_str(), CV_NODE_SEQ );
1125 cvStartWriteStruct( fs, 0, CV_NODE_MAP );
1126 tree->write( fs );
1127 cvEndWriteStruct( fs );
1130 cvEndWriteStruct( fs );
1133 cvEndWriteStruct( fs );
1142 void CvGBTrees::read( CvFileStorage* fs, CvFileNode* node )
1156 read_params( fs, node );
1161 base_value = (float)cvReadRealByName( fs, node, "base_value", 0.0 );
1162 class_count = cvReadIntByName( fs, node, "class_count", 1 );
1171 trees_fnode = cvGetFileNodeByName( fs, node, s.c_str() );
1188 CV_CALL(tree->read( fs, (CvFileNode*)reader.ptr, data ));
