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 vote146 views1 answer
How to count backwards in for loop python
Python
0 vote160 views2 answers
Django get all models
Python
-1 vote145 views1 answer
Python __new__
Python
3 votes149 views1 answer
Pandas loc for list
Python
3 votes153 views3 answers
Run flask in debug mode
Python
1 vote142 views1 answer
Python edit string variable
Python
0 vote256 views1 answer
Cuda memory in pytorch
Python
-3 votes147 views2 answers
Mean
Python
0 vote145 views1 answer
Creating a Pandas Data Frame Series
Python
0 vote170 views1 answer
Python download complete web page
Python