imputation.py

def Imputation(self, var ,stats=False):
        '''
        Function to Impute data using sklearn Imputer
        Input : Dataframe, variable(continuous/categorical)
        Output : None
        '''
        if var==self.continuous:
            imputer= SimpleImputer(strategy='median')
        if var==self.categorical:
            imputer = SimpleImputer(strategy='most_frequent')
        else:
            print("Imputation over continuous and categorical only")
        
        imputer.fit(self.data[var])
        self.data[var] = imputer.transform(self.data[var])
        print("Imputer Fitted and Transformed!")
        
        if stats==True:
            imputer.statistics_