lightgbm

Documentation

Index

• type Model
  • func NewModel() Model
  • func (model *Model) Load(modelfilename string) error
  • func (model Model) NumberOfLeaves() []int
  • func (model Model) Predict(fvals []float64, predLeaf bool) (float64, []uint32, error)

Types

type Model

type Model struct {
	// contains filtered or unexported fields
}

Model A trained model

func NewModel

func NewModel() Model

NewModel create a new model object

func (*Model) Load

func (model *Model) Load(modelfilename string) error

Load load the model from json file

func (Model) NumberOfLeaves

func (model Model) NumberOfLeaves() []int

NumberOfLeaves number of leaves for each estimator

func (Model) Predict

func (model Model) Predict(fvals []float64, predLeaf bool) (float64, []uint32, error)

Predict use model to predict result