问题背景
博主在调用sklearn.metrics.roc_auc_score()计算AUC时出现了这个报错
问题分析
错误提示说数据里面包含空值或者无穷大的数据,参与计算的两列数据有一列不是1就是0,另一列是楼主自己写的sigmoid函数计算出来的值,但是sigmoid函数不太可能算出无穷大的数值,所以关注是不是有哪个数出问题算出空值了。
- 先检查有哪些列存在空值
data = pd.read_csv('filename.csv')
print(data.isnull().any())
打印出来下面的结果:
date False
predict True
time