constant imputer function

constant_imputer.py

def ImputeWithConstant(train_df, test_df, cols=["taster_name"]):
  # function to impute taster_name
  # with 0 stands for "Unknown value"

  train_df = pd.DataFrame(train_df[cols])
  test_df = pd.DataFrame(test_df[cols])

  constant_imputer = SimpleImputer(strategy="constant", fill_value=0)

  constant_imputer.fit(train_df)
    
  imputed_train_set = constant_imputer.transform(train_df)
  imputed_train_df = pd.DataFrame(imputed_train_set, columns=train_df.columns)

  imputed_test_set = constant_imputer.transform(test_df)
  imputed_test_df = pd.DataFrame(imputed_test_set, columns=test_df.columns)

  return imputed_train_df, imputed_test_df