import fix

Author: SamoraHunter

ml_grid/util/impute_data_for_pipe.py

@@ -3,6 +3,7 @@
 import pandas as pd
 import numpy as np
 import random
+import pickle
 
 def mean_impute_dataframe(data, y_vars, test_size=0.25, val_size=0.25, random_state=1, seed=1234):
 
@@ -90,9 +91,6 @@ def mean_impute_dataframe(data, y_vars, test_size=0.25, val_size=0.25, random_st
 #df_merged = mean_impute_dataframe(df_merged, y_vars=outcome_columns, test_size=0.25, val_size=0.25, random_state=1, seed=1)
 
 
-import pandas as pd
-import pickle
-
 def save_missing_percentage(df, output_file='percent_missing.pkl'):
     """
     Calculate the percentage of missing values in a DataFrame and save it to a pickle file.