推荐系统实例之surprise库

surprise官方网址:http://surprise.readthedocs.io/en/stable/index.html

from surprise import KNNBasic
from surprise import Dataset
from surprise.model_selection import cross_validate

data = Dataset.load_builtin('ml-100k')
algo = KNNBasic()
cross_validate(algo, data, measures=['RMSE', 'MAE'], cv=3, verbose=True)

• 参数选择

from surprise import SVD
from surprise import Dataset
from surprise.model_selection import GridSearchCV

param_grid = {
   'n_epochs':[5