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

0 vote180 views1 answer
Calculate the R^2 for X and Y python
Python
1 vote190 views1 answer
Compute mfcc python
Python
4 votes204 views1 answer
Get current url python
Python
0 vote193 views1 answer
Python time limit for input
Python
3 votes208 views2 answers
Interpreter in python
Python
2 votes168 views1 answer
Print all keys having same value
Python
0 vote222 views1 answer
How to split text in a csv file in dataset python
Python
30 votes203 views3 answers
Iterate through all files in directory python
Python
0 vote190 views1 answer
Datediff in seconds in pandas
Python
-1 vote173 views2 answers
Model deployment with flask
Python Flask