在学习音乐推荐系统这个项目时,建模调参部分出现了问题
from surprise import SVD
from surprise import Dataset
from surprise import evaluate, print_perf
# 默认载入movielens数据集
data = Dataset.load_builtin('ml-100k')
# k折交叉验证(k=3)
data.split(n_folds=3)
# 试一把SVD矩阵分解
algo = SVD()
# 在数据集上测试一下效果
perf = evaluate(algo, data, measures=['RMSE', 'MAE', 'FCP'])
#输出结果
print_perf(perf)
报错提示:ImportError: cannot import name ‘evaluate’ from ‘surprise’ (D:\anaconda917\lib\site-packages\surprise_init_.py)
原因是:根据官方文档,evaluate()方法在1.0.5版中已弃用(功能上已由model_selection.cross_validate()取代),并在1.1.0版中删除(可能已安装)。
所以将from surprise import evaluate, print_perf
换成 from surprise.model_selection import cross_validate
即可。
from surprise import SVD
from surprise import Dataset
from surprise.model_selection import cross_validate
# Load the dataset (download it if needed)
data = Dataset.load_builtin('ml-100k')
# Use the famous SVD algorithm
algo = SVD()
# Run 5-fold cross-validation and then print results
cross_validate(algo, data, measures=['RMSE', 'MAE', 'FCP'], cv=5, verbose=True)