click

Aug 15, 2022

Changes in this version

type FitConfig

+ func (config *FitConfig) SetJobsAllocator(allocator *task.JobsAllocator) *FitConfig

Jul 9, 2022

Changes in this version

type FM

+ func (fm *FM) Complexity() int

type FactorizationMachine

+ Complexity func() int

type FitConfig

+ Task *task.Task

+ func (config *FitConfig) SetTask(t *task.Task) *FitConfig

type ModelSearcher

+ func (searcher *ModelSearcher) Complexity() int

May 25, 2022

Changes in this version

type Dataset

+ func (dataset *Dataset) Bytes() int

type FM

+ func (fm *FM) Bytes() int

type FactorizationMachine

+ Bytes func() int

type UnifiedDirectIndex

+ func (unified *UnifiedDirectIndex) Bytes() int

type UnifiedIndex

+ Bytes func() int

type UnifiedMapIndex

+ func (unified *UnifiedMapIndex) Bytes() int

Oct 16, 2021

Changes in this version

+ func MarshalIndex(w io.Writer, index UnifiedIndex) error

+ func MarshalModel(w io.Writer, m FactorizationMachine) error

type FM

+ func (fm *FM) Marshal(w io.Writer) error

+ func (fm *FM) Unmarshal(r io.Reader) error

type FactorizationMachine

+ func UnmarshalModel(r io.Reader) (FactorizationMachine, error)

+ Marshal func(w io.Writer) error

type UnifiedDirectIndex

+ func (unified *UnifiedDirectIndex) Marshal(w io.Writer) error

+ func (unified *UnifiedDirectIndex) Unmarshal(r io.Reader) error

type UnifiedIndex

+ func UnmarshalIndex(r io.Reader) (UnifiedIndex, error)

+ Marshal func(w io.Writer) error

+ Unmarshal func(r io.Reader) error

type UnifiedMapIndex

+ func (unified *UnifiedMapIndex) Marshal(w io.Writer) error

+ func (unified *UnifiedMapIndex) Unmarshal(r io.Reader) error

Sep 26, 2021

Changes in this version

type Dataset

+ CtxFeatures [][]int32

+ CtxValues [][]float32

+ ItemFeatures [][]int32

+ Items base.Integers

+ NormValues base.Floats

+ UserFeatures [][]int32

+ Users base.Integers

Sep 4, 2021

Changes in this version

+ func AUC(posPrediction, negPrediction []float32) float32

+ func Accuracy(posPrediction, negPrediction []float32) float32

+ func Precision(posPrediction, negPrediction []float32) float32

+ func Recall(posPrediction, _ []float32) float32

type Dataset

+ Features [][]int

+ NegativeCount int

+ PositiveCount int

+ Values [][]float32

type FitConfig

+ func (config *FitConfig) SetVerbose(verbose int) *FitConfig

type Score

+ AUC float32

+ Accuracy float32

+ Recall float32

+ func (score Score) ZapFields() []zap.Field

Aug 15, 2021

Changes in this version

type FitConfig

+ Tracker model.Tracker

+ func (config *FitConfig) SetTracker(tracker model.Tracker) *FitConfig

Jul 17, 2021

Changes in this version

+ func EncodeModel(m FactorizationMachine) ([]byte, error)

+ func LoadLibFMFile(path string) (labels [][]int, targets []float32, maxLabel int, err error)

+ type BaseFactorizationMachine struct

+ Index UnifiedIndex

+ func (b *BaseFactorizationMachine) Init(trainSet *Dataset)

+ type Dataset struct

+ Index UnifiedIndex

+ Inputs [][]int

+ Labels [][]int

+ Target []float32

+ func LoadDataFromBuiltIn(name string) (train, test *Dataset, err error)

+ func LoadDataFromDatabase(database data.Database, clickTypes []string, readType string) (*Dataset, error)

+ func (dataset *Dataset) Count() int

+ func (dataset *Dataset) Get(i int) ([]int, float32)

+ func (dataset *Dataset) ItemCount() int

+ func (dataset *Dataset) Split(ratio float32, seed int64) (*Dataset, *Dataset)

+ func (dataset *Dataset) UserCount() int

+ type FM struct

+ B float32

+ MaxTarget float32

+ MinTarget float32

+ Task FMTask

+ V [][]float32

+ W []float32

+ func NewFM(task FMTask, params model.Params) *FM

+ func (fm *FM) Clear()

+ func (fm *FM) Fit(trainSet, testSet *Dataset, config *FitConfig) Score

+ func (fm *FM) GetParamsGrid() model.ParamsGrid

+ func (fm *FM) Init(trainSet *Dataset)

+ func (fm *FM) InternalPredict(x []int) float32

+ func (fm *FM) Invalid() bool

+ func (fm *FM) Predict(userId, itemId string, itemLabels []string) float32

+ func (fm *FM) SetParams(params model.Params)

+ type FMTask string

+ const FMClassification

+ const FMRegression

+ type FactorizationMachine interface

+ Fit func(trainSet *Dataset, testSet *Dataset, config *FitConfig) Score

+ InternalPredict func(x []int) float32

+ Predict func(userId, itemId string, itemLabels []string) float32

+ func Clone(m FactorizationMachine) FactorizationMachine

+ func DecodeModel(buf []byte) (FactorizationMachine, error)

+ type FitConfig struct

+ Jobs int

+ Verbose int

+ func NewFitConfig() *FitConfig

+ func (config *FitConfig) LoadDefaultIfNil() *FitConfig

+ func (config *FitConfig) SetJobs(nJobs int) *FitConfig

+ type ModelSearcher struct

+ func NewModelSearcher(nEpoch, nTrials, nJobs int) *ModelSearcher

+ func (searcher *ModelSearcher) Fit(trainSet, valSet *Dataset) error

+ func (searcher *ModelSearcher) GetBestModel() (FactorizationMachine, Score)

+ func (searcher *ModelSearcher) IsClickModelHelpful() bool

+ type ParamsSearchResult struct

+ BestIndex int

+ BestModel FactorizationMachine

+ BestParams model.Params

+ BestScore Score

+ Params []model.Params

+ Scores []Score

+ func GridSearchCV(estimator FactorizationMachine, trainSet *Dataset, testSet *Dataset, ...) ParamsSearchResult

+ func RandomSearchCV(estimator FactorizationMachine, trainSet *Dataset, testSet *Dataset, ...) ParamsSearchResult

+ type Score struct

+ Precision float32

+ RMSE float32

+ Task FMTask

+ func EvaluateClassification(estimator FactorizationMachine, testSet *Dataset) Score

+ func EvaluateRegression(estimator FactorizationMachine, testSet *Dataset) Score

+ func (score Score) BetterThan(s Score) bool

+ func (score Score) GetName() string

+ func (score Score) GetValue() float32

+ type SnapshotManger struct

+ BestScore Score

+ BestWeights []interface{}

+ func (sm *SnapshotManger) AddSnapshot(score Score, weights ...interface{})

+ type UnifiedDirectIndex struct

+ N int

+ func (unified *UnifiedDirectIndex) CountContextLabels() int

+ func (unified *UnifiedDirectIndex) CountItemLabels() int

+ func (unified *UnifiedDirectIndex) CountItems() int

+ func (unified *UnifiedDirectIndex) CountUserLabels() int

+ func (unified *UnifiedDirectIndex) CountUsers() int

+ func (unified *UnifiedDirectIndex) EncodeContextLabel(label string) int

+ func (unified *UnifiedDirectIndex) EncodeItem(itemId string) int

+ func (unified *UnifiedDirectIndex) EncodeItemLabel(itemLabel string) int

+ func (unified *UnifiedDirectIndex) EncodeUser(userId string) int

+ func (unified *UnifiedDirectIndex) EncodeUserLabel(userLabel string) int

+ func (unified *UnifiedDirectIndex) GetContextLabels() []string

+ func (unified *UnifiedDirectIndex) GetItemLabels() []string

+ func (unified *UnifiedDirectIndex) GetItems() []string

+ func (unified *UnifiedDirectIndex) GetUserLabels() []string

+ func (unified *UnifiedDirectIndex) GetUsers() []string

+ func (unified *UnifiedDirectIndex) Len() int

+ type UnifiedIndex interface

+ CountContextLabels func() int

+ CountItemLabels func() int

+ CountItems func() int

+ CountUserLabels func() int

+ CountUsers func() int

+ EncodeContextLabel func(ctxLabel string) int

+ EncodeItem func(itemId string) int

+ EncodeItemLabel func(itemLabel string) int

+ EncodeUser func(userId string) int

+ EncodeUserLabel func(userLabel string) int

+ GetContextLabels func() []string

+ GetItemLabels func() []string

+ GetItems func() []string

+ GetUserLabels func() []string

+ GetUsers func() []string

+ Len func() int

+ func NewUnifiedDirectIndex(n int) UnifiedIndex

+ type UnifiedMapIndex struct

+ CtxLabelIndex base.Index

+ ItemIndex base.Index

+ ItemLabelIndex base.Index

+ UserIndex base.Index

+ UserLabelIndex base.Index

+ func (unified *UnifiedMapIndex) CountContextLabels() int

+ func (unified *UnifiedMapIndex) CountItemLabels() int

+ func (unified *UnifiedMapIndex) CountItems() int

+ func (unified *UnifiedMapIndex) CountUserLabels() int

+ func (unified *UnifiedMapIndex) CountUsers() int

+ func (unified *UnifiedMapIndex) EncodeContextLabel(label string) int

+ func (unified *UnifiedMapIndex) EncodeItem(itemId string) int

+ func (unified *UnifiedMapIndex) EncodeItemLabel(itemLabel string) int

+ func (unified *UnifiedMapIndex) EncodeUser(userId string) int

+ func (unified *UnifiedMapIndex) EncodeUserLabel(userLabel string) int

+ func (unified *UnifiedMapIndex) GetContextLabels() []string

+ func (unified *UnifiedMapIndex) GetItemLabels() []string

+ func (unified *UnifiedMapIndex) GetItems() []string

+ func (unified *UnifiedMapIndex) GetUserLabels() []string

+ func (unified *UnifiedMapIndex) GetUsers() []string

+ func (unified *UnifiedMapIndex) Len() int

+ type UnifiedMapIndexBuilder struct

+ CtxLabelIndex base.Index

+ ItemIndex base.Index

+ ItemLabelIndex base.Index

+ UserIndex base.Index

+ UserLabelIndex base.Index

+ func NewUnifiedMapIndexBuilder() *UnifiedMapIndexBuilder

+ func (builder *UnifiedMapIndexBuilder) AddCtxLabel(ctxLabel string)

+ func (builder *UnifiedMapIndexBuilder) AddItem(itemId string)

+ func (builder *UnifiedMapIndexBuilder) AddItemLabel(itemLabel string)

+ func (builder *UnifiedMapIndexBuilder) AddUser(userId string)

+ func (builder *UnifiedMapIndexBuilder) AddUserLabel(userLabel string)

+ func (builder *UnifiedMapIndexBuilder) Build() UnifiedIndex