python推荐算法库surprise_推荐系统实例的价格库,之,surprise

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,10], 'lr_all':[0.001,0.002,0.005], 'reg_all':[0.2,0.4,0.6]}

grid_search = GridSearchCV(SVD, param_grid, measures=['RMSE','FCP'], cv=3)

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

grid_search.fit(data)