Surprise
概述
surprise是一个便于使用的开源Python库,应用于推荐系统。
安装/使用
最简单的方式是使用pip:
$ pip install surprise
或者你可以clone仓库并build源:
$ git clone https://github.com/NicolasHug/surprise.git
$ python setup.py install
示例
下面是一个简单的例子,用户显示如何加载一个数据集:
from surprise import SVD
from surprise import Dataset
from surprise import evaluate
# Load the movielens-100k dataset (download it if needed),
# and split it into 3 folds for cross-validation.
data = Dataset.load_builtin("ml-100k")
data.split(n_folds=3)
# We"ll use the famous SVD algorithm.
algo = SVD()
# Evaluate performances of our algorithm on the dataset.
perf = evaluate(algo, data, measures=["RMSE", "MAE"])
print(perf)
输出:
Evaluating RMSE, MAE of algorithm SVD.
Fold 1 Fold 2 Fold 3 Mean
MAE 0.7475 0.7447 0.7425 0.7449
RMSE 0.9461 0.9436 0.9425 0.9441