I am writing a very basic program to predict missing values in a dataset using scikit-learn\'s Imputer class.
I have made a NumPy array, created an Imp
After scikit-learn version 0.20 the usage of impute module was changed. Now, we can use imputer like;
from sklearn.impute import SimpleImputer
impute = SimpleImputer(missing_values=np.nan, strategy='mean')
impute.fit(X)
X=impute.transform(X)
Pay attention:
Instead of 'NaN', np.nan is used
Don't need to use axis parameter
We can use imp or imputer instead of my impute
variable