推荐系统:ImportError: cannot import name ‘evaluate‘ from ‘surprise

根据官方文档,evaluate()方法在1.0.5版中已弃用(功能上已由model_selection.cross_validate()取代),并在1.1.0版中删除

分析:

需要改两个地方

第一个是导入库:将from surprise import evaluate 换成
from surprise.model_selection import cross_validate即可。

第二个是应用:data.split(n_folds=5)和evaluate(svd, data, measures=['RMSE', 'MAE'])两处代码不能用,要换成cross_validate函数。

具体如下:

原始代码:

from surprise import Reader, Dataset, SVD,evaluate
reader = Reader()
ratings = pd.read_csv('data/ratings_small.csv')
ratings.head()

data = Dataset.load_from_df(ratings[['userId', 'movieId', 'rating']], reader)
data.split(n_folds=5)
svd = SVD()
evaluate(svd, data, measures=['RMSE', 'MAE'])

更改后的代码:

from surprise import Reader, Dataset, SVD
from surprise.model_selection import cross_validate
reader = Reader()
ratings = pd.read_csv('data/ratings_small.csv')
ratings.head()

data = Dataset.load_from_df(ratings[['userId', 'movieId', 'rating']], reader)
# data.split(n_folds=5)
svd = SVD()
# evaluate(svd, data, measures=['RMSE', 'MAE'])
cross_validate(algo=svd,data=data,measures=['RMSE', 'MAE'],cv=5,verbose=True)

参考资料:

运行出错 ImportError:cannot import name 'evaluate' from 'surprise' - 灰信网(软件开发博客聚合)

 AttributeError: 'DatasetAutoFolds' object has no attribute 'split' - Stack Overflow

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值