解决sklearn.linear_model.LinearRegression模型ValueError: Input X contains NaN错误
这个错误是因为输入的自变量X包含NaN值。线性回归模型无法处理包含NaN值的数据。解决这个问题的方法有两种:
- 删除包含NaN值的样本:可以使用'dropna()'函数删除包含NaN值的行,然后再进行模型训练和预测。
data.dropna(inplace=True)
X = data[['信誉值均值', '预订任务限额均值', '商品周围商品个数', '会员点个数']]
y = data['任务标价']
model.fit(X, y)
- 使用数据填充NaN值:可以使用'fillna()'函数将NaN值替换为其他值,比如均值、中位数等。
X.fillna(X.mean(), inplace=True)
y.fillna(y.mean(), inplace=True)
model.fit(X, y)
根据实际情况选择合适的方法来处理NaN值。
原文地址: https://www.cveoy.top/t/topic/fzOs 著作权归作者所有。请勿转载和采集!