modify tree so that training is standalone

This commit is contained in:
tqchen
2014-02-26 16:03:00 -08:00
parent 2c6922f432
commit 4a612eb3ba
8 changed files with 197 additions and 287 deletions

View File

@@ -306,7 +306,7 @@ namespace xgboost{
}
};
};
namespace booster{
/*! \brief training parameters for regression tree */
struct TreeParamTrain{
@@ -431,5 +431,20 @@ namespace xgboost{
}
};
};
namespace booster{
/*! \brief node statistics used in regression tree */
struct RTreeNodeStat{
// loss chg caused by current split
float loss_chg;
// weight of current node
float base_weight;
// number of child that is leaf node known up to now
int leaf_child_cnt;
};
/*! \brief most comment structure of regression tree */
class RegTree: public TreeModel<bst_float,RTreeNodeStat>{
};
};
};
#endif