yuenshingyan/MissForest

Import MissForest / Typo in readme

nwieder opened this issue · 2 comments

importing MissForest with:

from missforest.missforest import

Leads to an Error:

`---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
Cell In[2], line 23
15 from sklearn.preprocessing import StandardScaler, OrdinalEncoder, OneHotEncoder, LabelEncoder
17 # Small fix to make missingpy forward compatible
18 # import sklearn.neighbors._base
19 # sys.modules['sklearn.neighbors.base'] = sklearn.neighbors._base
20
21 # from missingpy import MissForest
---> 23 from missforest.missforest import MissForest

ModuleNotFoundError: No module named 'missforest.missforest'
`

The correct way to import MissForest is:

from missforest.miss_forest import

Below code shall solve the issue:

import sklearn.neighbors._base # this is from `scikit-learn` instead of `sklearn`
import sys
sys.modules['sklearn.neighbors.base'] = sklearn.neighbors._base
from missingpy import MissForest # remember to put this after sys.modules