推荐算法 Surprise（一）

自动交叉验证：

Surprise有很多内置算法和数据集供你使用。最简单的，我们可以只用几行代码运行交叉验证程序。

from surprise import SVD
from surprise import Dataset
from surprise.model_selection import cross_validate

date = Dataset.load_builtin('ml-100k')

algo = SVD()

cross_validate(algo,date,measures=['RMSE', 'MAE'],cv=5,verbose=True)

运行结果如下

Dataset ml-100k could not be found. Do you want to download it? [Y/n] y
Trying to download dataset from http://files.grouplens.org/datasets/movielens/ml-100k.zip...
Done! Dataset ml-100k has been saved to C:\Users\60191/.surprise_data/ml-100k
Evaluating RMSE, MAE of algorithm SVD on 5 split(s).

                  Fold 1  Fold 2  Fold 3  Fold 4  Fold 5  Mean    Std     
RMSE (testset)    0.9422  0.9361  0.9327  0.9296  0.9396  0.9360  0.0045  
MAE (testset)     0.7423  0.7390  0.7351  0.7324  0.7399  0.7377  0.0035  
Fit time          4.42    4.70    4.53    4.68