Nearest neighbors imputation

# Nearest neighbors imputation

import numpy as np
from sklearn.impute import KNNImputer
nan = np.nan
X = [[1, 2, nan], [3, 4, 3], [nan, 6, 5], [8, 8, 7]]
imputer = KNNImputer(n_neighbors=2, weights="uniform")
imputer.fit_transform(X)
# array([[1. , 2. , 4. ],
#        [3. , 4. , 3. ],
#        [5.5, 6. , 5. ],
#        [8. , 8. , 7. ]])

Source: scikit-learn.org

Similar codes

1 vote121 views1 answer
Keras import optimizer adam
Python
1 vote101 views1 answer
Django get form id from request
Python
0 vote97 views1 answer
Read dict from text
Python
0 vote114 views1 answer
Stemming words python
Python
1 vote106 views1 answer
Append to list in dictionary python if exists
Python
0 vote134 views2 answers
ImportError: /usr/local/lib/python3.7/dist-packages/cv2/cv2.cpython-37m-arm-linux-gnueabihf.so: undefined symbol: __atomic_fetch_add_8
Python
10 votes110 views6 answers
Variable in python
Python
1 vote94 views1 answer
Count missing values groupby
Python
0 vote108 views2 answers
Push button raspberry pi
Python
-1 vote101 views2 answers
Python check if string has space
Python