MICE和ICkNNI填充案例

 

 

 

 

import numpy as np
import pandas as pd
import miceforest as mf
import matplotlib.pyplot as plt
from sklearn.impute import KNNImputer
from sklearn.experimental import enable_iterative_imputer
from sklearn.impute import IterativeImputer
from sklearn.ensemble import RandomForestRegressor
from sklearn.linear_model import LogisticRegression


x1 = np.random.uniform(0,10,800)
x2 = np.random.uniform(0,10,800)

n1 = np.linspace(0,5,1000)
m1 = np.sqrt(25 - n1**2)

n2 = np.linspace(0,9,1000)
m2 = np.sqrt(81 - n2**2)

X = np.vstack((x1,x2)).T
y = np.ones(800)
for i in