xgjson

type GBLinearBooster struct {
	Model xgbin.GBLinearModel `json:"model"`
	Name  string              `json:"name"`
}

type GBLinearJson struct {
	Learner GBLinearLearner `json:"learner"`
	Version []int           `json:"version"`
}

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 struct {
	Model      GBTreeModel `json:"model"`
	WeightDrop []float64   `json:"weight_drop"`
	Name       string      `json:"name"`
}

type GBTreeJson struct {
	Learner GBTreeLearner `json:"learner"`
	Version []int         `json:"version"`
}

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 struct {
	GbTreeModelParam xgbin.GBTreeModelParam `json:"gbtree_model_param"`
	Trees            []*Tree                `json:"trees"`
	TreeInfo         []int32                `json:"tree_info"`
}

type Objective struct {
	Name         string       `json:"name"`
	RegLossParam RegLossParam `json:"reg_loss_param"`
}

type RegLossParam struct {
	ScalePosWeight string `json:"scale_pos_weight"`
}

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"`
}