Python中用于推荐系统的一些开源库

最新推荐文章于 2025-06-26 11:24:44 发布

原创

最新推荐文章于 2025-06-26 11:24:44 发布 · 3.7k 阅读

CC 4.0 BY-SA版权

本文介绍了Python中的两个推荐系统库：surprise和LightFM。surprise是一个常用的推荐系统库，支持多种协同过滤算法和评估标准；LightFM则是一个用于隐式和显式反馈的推荐算法库，支持将元数据纳入传统矩阵分解，并提供了多种损失函数。文章详细阐述了这两个库的使用方法和关键特性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近新发现一个推荐系统库lightfm，为了防止忘记，和之前用过的surprise库一起做个小笔记。

surprise可以算上是推荐系统中较为常用，知名度较高的库了，它是是scikit系列中的一个库。
官方文档：https://surprise.readthedocs.io/en/stable/getting_started.html
github：https://github.com/NicolasHug/Surprise
surprise库支持多种推荐算法

algorithm	describe
random_pred.NormalPredictor	Algorithm predicting a random rating based on the distribution of the training set, which is assumed to be normal.
baseline_only.BaselineOnly	Algorithm predicting the baseline estimate for given user and item.
knns.KNNBasic	A basic collaborative filtering algorithm.
knns.KNNWithMeans	A basic collaborative filtering algorithm, taking into account the mean ratings of each user.
knns.KNNWithZScore	A basic collaborative filtering algorithm, taking into account the z-score normalization of each user.
knns.KNNBaseline	A basic collaborative filtering algorithm taking into account a baseline rating.
matrix_factorization.SVD	The famous SVD algorithm, as popularized by Simon Funk during the Netflix Prize.