class DataFrame_Categorical_Imputer(): def __init__(self): print("Imputation object created") def fit(self, data): """ This method will fit impute mode value for all missing categoriical variables """ self.fill = pd.Series([data[column].\ value_counts().index[0] if data[column].dtype == np.dtype('O') else \ data[column].mode() for column in data], index=data.columns) return self