Documentation ¶
Index ¶
- type Dataset
- type FitInfo
- type Id
- type Option
- type Rec
- type Recommender
- func FitEvalExplicit[T Id, U Id](trainSet *Dataset[T, U], validSet *Dataset[T, U], options ...Option) (*Recommender[T, U], error)
- func FitExplicit[T Id, U Id](trainSet *Dataset[T, U], options ...Option) (*Recommender[T, U], error)
- func FitImplicit[T Id, U Id](trainSet *Dataset[T, U], options ...Option) (*Recommender[T, U], error)
- func (r *Recommender[T, U]) GlobalMean() float32
- func (r *Recommender[T, U]) ItemFactors(itemId U) []float32
- func (r *Recommender[T, U]) ItemIds() []U
- func (r *Recommender[T, U]) ItemRecs(itemId U, count int) []Rec[U]
- func (r *Recommender[T, U]) Predict(userId T, itemId U) float32
- func (r *Recommender[T, U]) Rmse(data *Dataset[T, U]) float32
- func (r *Recommender[T, U]) SimilarUsers(userId T, count int) []Rec[T]
- func (r *Recommender[T, U]) UserFactors(userId T) []float32
- func (r *Recommender[T, U]) UserIds() []T
- func (r *Recommender[T, U]) UserRecs(userId T, count int) []Rec[U]
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
This section is empty.
Types ¶
type Dataset ¶
func NewDataset ¶
type Recommender ¶
func FitEvalExplicit ¶
func FitExplicit ¶
func FitImplicit ¶
func (*Recommender[T, U]) GlobalMean ¶
func (r *Recommender[T, U]) GlobalMean() float32
func (*Recommender[T, U]) ItemFactors ¶
func (r *Recommender[T, U]) ItemFactors(itemId U) []float32
func (*Recommender[T, U]) ItemIds ¶
func (r *Recommender[T, U]) ItemIds() []U
func (*Recommender[T, U]) ItemRecs ¶
func (r *Recommender[T, U]) ItemRecs(itemId U, count int) []Rec[U]
func (*Recommender[T, U]) Predict ¶
func (r *Recommender[T, U]) Predict(userId T, itemId U) float32
func (*Recommender[T, U]) Rmse ¶
func (r *Recommender[T, U]) Rmse(data *Dataset[T, U]) float32
func (*Recommender[T, U]) SimilarUsers ¶
func (r *Recommender[T, U]) SimilarUsers(userId T, count int) []Rec[T]
func (*Recommender[T, U]) UserFactors ¶
func (r *Recommender[T, U]) UserFactors(userId T) []float32
func (*Recommender[T, U]) UserIds ¶
func (r *Recommender[T, U]) UserIds() []T
func (*Recommender[T, U]) UserRecs ¶
func (r *Recommender[T, U]) UserRecs(userId T, count int) []Rec[U]
Click to show internal directories.
Click to hide internal directories.