fm

package
v0.0.0-...-14a59ba Latest Latest
Warning

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

Go to latest
Published: Oct 27, 2017 License: MIT Imports: 3 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Commands

func Commands() []cli.Command

Types

type FactorizeMachine

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

func (*FactorizeMachine) Clear

func (c *FactorizeMachine) Clear()

func (*FactorizeMachine) Command

func (c *FactorizeMachine) Command() cli.Command

func (*FactorizeMachine) Init

func (c *FactorizeMachine) Init(ctx *cli.Context)

func (*FactorizeMachine) LoadModel

func (self *FactorizeMachine) LoadModel(path string)

func (*FactorizeMachine) Predict

func (c *FactorizeMachine) Predict(sample *core.Sample) float64

func (*FactorizeMachine) SaveModel

func (self *FactorizeMachine) SaveModel(path string)

func (*FactorizeMachine) Train

func (c *FactorizeMachine) Train(dataset *core.DataSet)

type FactorizeMachineParams

type FactorizeMachineParams struct {
	LearningRate   float64
	Regularization float64
	FactorNumber   int
}

Jump to

Keyboard shortcuts

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