gradientboostedtrees

package
v0.0.0-...-7d47eef Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Apr 12, 2023 License: Apache-2.0, MIT Imports: 10 Imported by: 0

Documentation

Overview

Package gradientboostedtrees defines the gradient boosted trees model.

Index

Constants

View Source
const ModelKey = "GRADIENT_BOOSTED_TREES"

ModelKey is the unique identifier of the model for serizalization.

Variables

This section is empty.

Functions

func Create

Create creates a GBT model.

Types

type Model

type Model struct {
	GbtHeader *gbt_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.

Directories

Path Synopsis
Package proto includes all proto definitions used in the golang package in one large package.
Package proto includes all proto definitions used in the golang package in one large package.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL