def auto_impute_df()

in src/utils.py [0:0]

• 8 lines of code
• 5 McCabe index (conditional complexity)

def auto_impute_df(df):
    for i in df.columns:
        if df[i].dtype == 'object':
            df[i] = df[i].fillna(df[i].mode().iloc[0])
            df[i] = le.fit_transform(df[i])
        elif df[i].dtype == 'int' or df[i].dtype == 'float':
            df[i] = df[i].fillna(np.mean(df[i]))

    return df