Documentation ¶
Index ¶
- func BinarizeLabel(rating float64) float64
- func BinarizeLabel32(rating float32) float32
- func PreFillUbCache(ubc *ubcache.UserBehaviorCache, table string) (err error)
- type MovielensRec
- func (recSys *MovielensRec) GetDashboardOverview(ctx context.Context) (res rcmd.DashboardOverviewResult, err error)
- func (recSys *MovielensRec) GetItemFeature(ctx context.Context, itemId int) (tensor rcmd.Tensor, err error)
- func (recSys *MovielensRec) GetItemsFeatureOverview(ctx context.Context, offset, size int, _ map[string][]string) (res rcmd.ItemOverviewResult, err error)
- func (recSys *MovielensRec) GetUserBehavior(ctx context.Context, userId int, maxLen int64, maxPk int64, maxTs int64) (itemSeq []int, err error)
- func (recSys *MovielensRec) GetUserFeature(ctx context.Context, userId int) (tensor rcmd.Tensor, err error)
- func (recSys *MovielensRec) GetUsersFeatureOverview(ctx context.Context, offset, size int, _ map[string][]string) (res rcmd.UserItemOverviewResult, err error)
- func (recSys *MovielensRec) ItemSeqGenerator(ctx context.Context) (ret <-chan string, err error)
- func (recSys *MovielensRec) PreRank(ctx context.Context) (err error)
- func (recSys *MovielensRec) PreTrain(ctx context.Context) (err error)
- func (recSys *MovielensRec) SampleGenerator(_ context.Context) (ret <-chan rcmd.Sample, err error)
- type YoutubeDnnImpl
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func BinarizeLabel ¶
func BinarizeLabel32 ¶ added in v0.3.0
func PreFillUbCache ¶ added in v0.2.0
func PreFillUbCache(ubc *ubcache.UserBehaviorCache, table string) (err error)
PreFillUbCache prefill ubcache with data from db `ub_test` or `ub_train`. the ub_train table is generated by SQL like:
```sql create table ratings_train_desc as select r.userId, movieId, rating, timestamp from ratings_train r order by r.userId, timestamp desc; create table ub_train as select userId, group_concat(movieId) movieIds ,group_concat(timestamp) timestamps from ratings_train_desc group by userId order by timestamp; ```
Sample from ub_train will be like:
31699, "246,247,252,260,265", "825638410,825638407,825638403,825638401,825638400"
Types ¶
type MovielensRec ¶ added in v0.2.0
type MovielensRec struct { DataPath string SampleCnt int // contains filtered or unexported fields }
func (*MovielensRec) GetDashboardOverview ¶ added in v0.2.0
func (recSys *MovielensRec) GetDashboardOverview(ctx context.Context) (res rcmd.DashboardOverviewResult, err error)
func (*MovielensRec) GetItemFeature ¶ added in v0.2.0
func (*MovielensRec) GetItemsFeatureOverview ¶ added in v0.2.0
func (recSys *MovielensRec) GetItemsFeatureOverview(ctx context.Context, offset, size int, _ map[string][]string) (res rcmd.ItemOverviewResult, err error)
func (*MovielensRec) GetUserBehavior ¶ added in v0.2.0
func (*MovielensRec) GetUserFeature ¶ added in v0.2.0
func (*MovielensRec) GetUsersFeatureOverview ¶ added in v0.2.0
func (recSys *MovielensRec) GetUsersFeatureOverview(ctx context.Context, offset, size int, _ map[string][]string) (res rcmd.UserItemOverviewResult, err error)
func (*MovielensRec) ItemSeqGenerator ¶ added in v0.2.0
func (recSys *MovielensRec) ItemSeqGenerator(ctx context.Context) (ret <-chan string, err error)
func (*MovielensRec) PreRank ¶ added in v0.2.0
func (recSys *MovielensRec) PreRank(ctx context.Context) (err error)
PreRank is called before rank, it can be used to prefill ub cache.
func (*MovielensRec) PreTrain ¶ added in v0.2.0
func (recSys *MovielensRec) PreTrain(ctx context.Context) (err error)
func (*MovielensRec) SampleGenerator ¶ added in v0.2.0
type YoutubeDnnImpl ¶ added in v0.4.0
type YoutubeDnnImpl struct {
// contains filtered or unexported fields
}
func (*YoutubeDnnImpl) Fit ¶ added in v0.4.0
func (d *YoutubeDnnImpl) Fit(trainSample *rcmd.TrainSample) (pred rcmd.PredictAbstract, err error)
Click to show internal directories.
Click to hide internal directories.