【发布时间】:2019-02-24 05:42:42
【问题描述】:
我正在使用 scikit-learn 库学习机器学习,当我看到来自Scikit-learn tutorial的代码时,我感到很困惑
代码如下:
import numpy as np
from sklearn.impute import SimpleImputer
imp = SimpleImputer(missing_values=np.nan, strategy='mean')
imp.fit([[1, 2], [np.nan, 3], [7, 6]])
X = [[np.nan, 2], [6, np.nan], [7, 6]]
print(imp.transform(X))
据我了解, imputer 用于填充缺失值,并且它使用某种策略,例如均值、中值或众数。但是我不明白这3行代码在做什么
imp.fit([[1, 2], [np.nan, 3], [7, 6]])
X = [[np.nan, 2], [6, np.nan], [7, 6]]
print(imp.transform(X))
为什么它适合带有 np.nan 的矩阵?以及这些拟合如何影响imp.transform(X)?
【问题讨论】:
标签: python scikit-learn