modify tree so that training is standalone
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user