randomforest

Documentation

Overview

Package randomforest defines the gradient boosted trees model.

Index

• Constants
• func Create(header *model_pb.AbstractModel, dataspec *dataspec_pb.DataSpecification) model.Implementation
• type Model
  • func (me *Model) Dataspec() *dataspec_pb.DataSpecification
  • func (me *Model) Header() *model_pb.AbstractModel
  • func (me *Model) LoadSpecific(modelPath string, prefix string) error
  • func (me *Model) Name() string

Constants

const ModelKey = "RANDOM_FOREST"

ModelKey is the unique identifier of the model for serizalization.

Functions

func Create

func Create(header *model_pb.AbstractModel, dataspec *dataspec_pb.DataSpecification) model.Implementation

Create creates a GBT model.

Types

type Model

type Model struct {
	RfHeader *rf_pb.Header
	Forest   *dt.Forest
	// contains filtered or unexported fields
}

Model is a Gradient Boosted Trees model.

func (*Model) Dataspec

func (me *Model) Dataspec() *dataspec_pb.DataSpecification

Dataspec of the model.

func (*Model) Header

func (me *Model) Header() *model_pb.AbstractModel

Header of the model.

func (*Model) LoadSpecific

func (me *Model) LoadSpecific(modelPath string, prefix string) error

LoadSpecific loads a model from disk.

func (*Model) Name

func (me *Model) Name() string

Name of the model.