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

2 votes168 views6 answers
Variable python
Python
0 vote158 views1 answer
Python array set key value
Python
18 votes175 views2 answers
Absolute value python
Python
4 votes159 views2 answers
Check if a string is float python
Python
0 vote162 views1 answer
Z transform in python
Python
0 vote157 views1 answer
Compile and train cnn models
Python
8 votes169 views4 answers
Remove nans from array
Python
0 vote159 views1 answer
Real hour in python
Python
49 votes179 views3 answers
Python write csv line by line
Python
0 vote162 views1 answer
How to convert response to beautifulsoup object
Python