Documentation ¶
Index ¶
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type GBLinearBooster ¶
type GBLinearBooster struct { Model xgbin.GBLinearModel `json:"model"` Name string `json:"name"` }
type GBLinearJson ¶
type GBLinearJson struct { Learner GBLinearLearner `json:"learner"` Version []int `json:"version"` }
func ReadGBLinear ¶
func ReadGBLinear(filePath string) (*GBLinearJson, error)
type GBLinearLearner ¶
type GBLinearLearner struct { FeatureNames []string `json:"feature_names"` FeatureTypes []string `json:"feature_types"` GradientBooster GBLinearBooster `json:"gradient_booster"` Objective Objective `json:"objective"` LearnerModelParam xgbin.LearnerModelParamLegacy `json:"learner_model_param"` }
type GBTreeBooster ¶
type GBTreeBooster struct { Model GBTreeModel `json:"model"` WeightDrop []float64 `json:"weight_drop"` Name string `json:"name"` }
type GBTreeJson ¶
type GBTreeJson struct { Learner GBTreeLearner `json:"learner"` Version []int `json:"version"` }
func ReadGBTree ¶
func ReadGBTree(filePath string) (*GBTreeJson, error)
type GBTreeLearner ¶
type GBTreeLearner struct { FeatureNames []string `json:"feature_names"` FeatureTypes []string `json:"feature_types"` GradientBooster GBTreeBooster `json:"gradient_booster"` Objective Objective `json:"objective"` LearnerModelParam xgbin.LearnerModelParamLegacy `json:"learner_model_param"` }
type GBTreeModel ¶
type GBTreeModel struct { GbTreeModelParam xgbin.GBTreeModelParam `json:"gbtree_model_param"` Trees []*Tree `json:"trees"` TreeInfo []int32 `json:"tree_info"` }
func (*GBTreeModel) ToBinGBTreeModel ¶
func (g *GBTreeModel) ToBinGBTreeModel() *xgbin.GBTreeModel
type Objective ¶
type Objective struct { Name string `json:"name"` RegLossParam RegLossParam `json:"reg_loss_param"` }
type RegLossParam ¶
type RegLossParam struct {
ScalePosWeight string `json:"scale_pos_weight"`
}
type Tree ¶
type Tree struct { TreeParam xgbin.TreeParam `json:"tree_param"` Id int `json:"id"` LossChanges []float32 `json:"loss_changes"` SumHessian []float32 `json:"sum_hessian"` BaseWeights []float32 `json:"base_weights"` LeftChildren []int32 `json:"left_children"` RightChildren []int32 `json:"right_children"` Parents []int32 `json:"parents"` SplitIndices []uint32 `json:"split_indices"` SplitConditions []float32 `json:"split_conditions"` SplitType []int32 `json:"split_type"` DefaultLeft []bool `json:"default_left"` Categories []int32 `json:"categories"` CategoriesNodes []int32 `json:"categories_nodes"` CategoriesSegments []int32 `json:"categories_segments"` CategoricalSizes []int32 `json:"categorical_sizes"` }
Click to show internal directories.
Click to hide internal directories.