Practical demonstration of SimpleImputer

KNNImputer_basic.py

# import the required library
from sklearn.impute import KNNImputer

# define imputer
imputer = KNNImputer() # you might want to try different n_neighbors parameters. Default is 5

# fit on the training dataset
imputer.fit(X_train)

# transform the training dataset (don't forget to transform the test dataset as well)
X_train_clean = pd.DataFrame(imputer.transform(X_train), columns = X_train.columns, index = X_train.index)