recommenders入门学习资料 - 推荐系统模型构建工具库

Ray

recommenders

recommenders简介

recommenders是一个开源的推荐系统工具库,由微软开发并维护,现隶属于Linux Foundation AI & Data基金会。它提供了构建推荐系统的全流程支持,包括:

数据准备
模型构建
模型评估
超参数调优
模型部署

该库的目标是帮助研究人员、开发者和爱好者快速原型化、实验和生产各种经典和前沿的推荐算法。

recommenders logo

主要特性

recommenders具有以下主要特性:

提供多种经典和最新推荐算法的实现
支持CPU、GPU和Spark等多种计算环境
包含数据处理、特征工程等实用工具函数
提供标准化的评估指标和方法
易于与生产环境集成
详细的文档和示例notebook

快速开始

要开始使用recommenders,可以通过pip安装:

pip install recommenders

然后可以导入并使用其中的算法和工具:

import tensorflow_datasets as tfds
import tensorflow_recommenders as tfrs

# 加载数据
ratings = tfds.load("movielens/100k-ratings", split="train")
movies = tfds.load("movielens/100k-movies", split="train")

# 构建模型
user_model = tf.keras.Sequential([...])
movie_model = tf.keras.Sequential([...])

# 定义任务
task = tfrs.tasks.Retrieval(
  metrics=tfrs.metrics.FactorizedTopK(
    movies.batch(128).map(movie_model)
  )
)

# 训练模型
model = MovielensModel(user_model, movie_model, task)
model.compile(optimizer=tf.keras.optimizers.Adagrad(0.5))
model.fit(ratings.batch(4096), epochs=3)

# 获取推荐
index = tfrs.layers.ann.BruteForce(model.user_model)
index.index_from_dataset(
  movies.batch(100).map(lambda title: (title, model.movie_model(title)))
)
_, titles = index(np.array(["42"]))
print(f"Recommendations for user 42: {titles[0, :3]}")