python - MissForest 不适用于分类变量

Question

我正在为我的混合类型数据实现 Missforest（使用 Scikit-Learn），但它不适用于我的分类特征。例如，将 1.6 替换为 Gender 而不是 1 或 0，并且不计算分类变量的众数。我在 python 中使用了这些行。

import sklearn.neighbors._base
sys.modules['sklearn.neighbors.base'] = sklearn.neighbors._base
from missingpy import MissForest
Imputer = MissForest()
X_imputed = imputer.fit_transform(data)

标签： pythonscikit-learn