Lines Matching refs:fs
1532 void DTreesImpl::writeTrainingParams(FileStorage& fs) const
1534 fs << "use_surrogates" << (params.useSurrogates ? 1 : 0);
1535 fs << "max_categories" << params.getMaxCategories();
1536 fs << "regression_accuracy" << params.getRegressionAccuracy();
1538 fs << "max_depth" << params.getMaxDepth();
1539 fs << "min_sample_count" << params.getMinSampleCount();
1540 fs << "cross_validation_folds" << params.getCVFolds();
1543 fs << "use_1se_rule" << (params.use1SERule ? 1 : 0);
1546 fs << "priors" << params.priors;
1549 void DTreesImpl::writeParams(FileStorage& fs) const
1551 fs << "is_classifier" << isClassifier();
1552 fs << "var_all" << (int)varType.size();
1553 fs << "var_count" << getVarCount();
1562 fs << "ord_var_count" << ord_var_count;
1563 fs << "cat_var_count" << cat_var_count;
1565 fs << "training_params" << "{";
1566 writeTrainingParams(fs);
1568 fs << "}";
1572 fs << "global_var_idx" << 1;
1573 fs << "var_idx" << varIdx;
1576 fs << "var_type" << varType;
1579 fs << "cat_ofs" << catOfs;
1581 fs << "cat_map" << catMap;
1583 fs << "class_labels" << classLabels;
1585 fs << "missing_subst" << missingSubst;
1588 void DTreesImpl::writeSplit( FileStorage& fs, int splitidx ) const
1592 fs << "{:";
1595 fs << "var" << vi;
1596 fs << "quality" << split.quality;
1609 fs << (default_dir*(split.inversed ? -1 : 1) > 0 ? "in" : "not_in") << "[:";
1615 fs << i;
1618 fs << "]";
1621 fs << (!split.inversed ? "le" : "gt") << split.c;
1623 fs << "}";
1626 void DTreesImpl::writeNode( FileStorage& fs, int nidx, int depth ) const
1629 fs << "{";
1630 fs << "depth" << depth;
1631 fs << "value" << node.value;
1634 fs << "norm_class_idx" << node.classIdx;
1638 fs << "splits" << "[";
1641 writeSplit( fs, splitidx );
1643 fs << "]";
1646 fs << "}";
1649 void DTreesImpl::writeTree( FileStorage& fs, int root ) const
1651 fs << "nodes" << "[";
1661 writeNode( fs, nidx, depth );
1679 fs << "]";
1682 void DTreesImpl::write( FileStorage& fs ) const
1684 writeParams(fs);
1685 writeTree(fs, roots[0]);
