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 vote290 views1 answer
Save lines from a file
Python
10 votes291 views2 answers
How to detect an even number in python
Python
0 vote286 views2 answers
Python list to string with spaces
Python
0 vote277 views1 answer
Code runner runs python 2
Python
0 vote290 views1 answer
Django delete session
Python
0 vote325 views1 answer
Restaurant management system in django
Python Django
127 votes326 views6 answers
Pandas rename column name
Python
0 vote330 views1 answer
Import data in pandad
Python
0 vote298 views1 answer
How to disconnect wifi using python
Python
4 votes340 views2 answers
Replace nan using fillna
Python