Hardcode columns to keep.

steenrotsman · May 13, 2024 · 56133e3 · 56133e3
1 parent 6cb14f9
commit 56133e3
Showing 1 changed file with 210 additions and 3 deletions.
diff --git a/submission.py b/submission.py
@@ -34,13 +34,14 @@ def clean_df(df, background_df=None):
     pd.DataFrame: The cleaned dataframe with only the necessary columns and processed variables.
     """
     max_waves = background_df.groupby("nomem_encr")["wave"].max().reset_index()
-    return (
+    clean = (
         df.merge(max_waves, on="nomem_encr")
         .merge(background_df, on=["nomem_encr", "wave"])
-        .dropna(axis=1, thresh=len(df) * 0.4)
         .fillna(-1)
     )
 
+    return clean[cols]
+
 
 def predict_outcomes(df, background_df=None, model_path="model.joblib"):
     """Generate predictions using the saved model and the input dataframe.
@@ -85,7 +86,213 @@ def predict_outcomes(df, background_df=None, model_path="model.joblib"):
     )
 
     # Combine predictions for individual
-    df_predict = (df_predict["prediction"] > THRESHOLD).astype(int)
+    df_predict["prediction"] = (df_predict["prediction"] > THRESHOLD).astype(int)
 
     # Return only dataset with predictions and identifier
     return df_predict
+
+
+cols = [
+    "nomem_encr",
+    "outcome_available",
+    "birthyear_bg",
+    "gender_bg",
+    "migration_background_bg",
+    "age_bg",
+    "belbezig_2008",
+    "belbezig_2009",
+    "belbezig_2010",
+    "belbezig_2011",
+    "belbezig_2012",
+    "belbezig_2013",
+    "belbezig_2014",
+    "belbezig_2015",
+    "belbezig_2016",
+    "belbezig_2017",
+    "belbezig_2018",
+    "belbezig_2019",
+    "belbezig_2020",
+    "brutohh_f_2009",
+    "brutohh_f_2010",
+    "brutohh_f_2011",
+    "brutohh_f_2014",
+    "brutohh_f_2015",
+    "brutohh_f_2017",
+    "brutohh_f_2018",
+    "brutoink_f_2008",
+    "brutoink_f_2009",
+    "brutoink_f_2010",
+    "brutoink_f_2011",
+    "brutoink_f_2012",
+    "brutoink_f_2013",
+    "brutoink_f_2014",
+    "brutoink_f_2015",
+    "brutoink_f_2017",
+    "brutoink_f_2018",
+    "brutoink_f_2020",
+    "burgstat_2008",
+    "burgstat_2009",
+    "burgstat_2010",
+    "burgstat_2011",
+    "burgstat_2012",
+    "burgstat_2013",
+    "burgstat_2014",
+    "burgstat_2015",
+    "burgstat_2016",
+    "burgstat_2017",
+    "burgstat_2018",
+    "burgstat_2019",
+    "burgstat_2020",
+    "nettohh_f_2009",
+    "nettohh_f_2010",
+    "nettohh_f_2011",
+    "nettohh_f_2014",
+    "nettohh_f_2015",
+    "nettohh_f_2017",
+    "nettohh_f_2018",
+    "nettoink_2008",
+    "nettoink_2009",
+    "nettoink_2010",
+    "nettoink_2011",
+    "nettoink_2012",
+    "nettoink_2013",
+    "nettoink_2014",
+    "nettoink_2015",
+    "nettoink_2017",
+    "nettoink_2018",
+    "nettoink_2020",
+    "nettoink_f_2008",
+    "nettoink_f_2009",
+    "nettoink_f_2010",
+    "nettoink_f_2011",
+    "nettoink_f_2012",
+    "nettoink_f_2013",
+    "nettoink_f_2014",
+    "nettoink_f_2015",
+    "nettoink_f_2017",
+    "nettoink_f_2018",
+    "nettoink_f_2020",
+    "oplcat_2008",
+    "oplcat_2009",
+    "oplcat_2010",
+    "oplcat_2011",
+    "oplcat_2012",
+    "oplcat_2013",
+    "oplcat_2014",
+    "oplcat_2015",
+    "oplcat_2016",
+    "oplcat_2017",
+    "oplcat_2018",
+    "oplcat_2019",
+    "oplcat_2020",
+    "oplmet_2008",
+    "oplmet_2009",
+    "oplmet_2010",
+    "oplmet_2011",
+    "oplmet_2012",
+    "oplmet_2013",
+    "oplmet_2014",
+    "oplmet_2015",
+    "oplmet_2016",
+    "oplmet_2017",
+    "oplmet_2018",
+    "oplmet_2019",
+    "oplmet_2020",
+    "oplzon_2008",
+    "oplzon_2009",
+    "oplzon_2010",
+    "oplzon_2011",
+    "oplzon_2012",
+    "oplzon_2013",
+    "oplzon_2014",
+    "oplzon_2015",
+    "oplzon_2016",
+    "oplzon_2017",
+    "oplzon_2018",
+    "oplzon_2019",
+    "oplzon_2020",
+    "partner_2008",
+    "partner_2009",
+    "partner_2010",
+    "partner_2011",
+    "partner_2012",
+    "partner_2013",
+    "partner_2014",
+    "partner_2015",
+    "partner_2016",
+    "partner_2017",
+    "partner_2018",
+    "partner_2019",
+    "partner_2020",
+    "sted_2008",
+    "sted_2009",
+    "sted_2010",
+    "sted_2011",
+    "sted_2012",
+    "sted_2013",
+    "sted_2014",
+    "sted_2015",
+    "sted_2016",
+    "sted_2017",
+    "sted_2018",
+    "sted_2019",
+    "sted_2020",
+    "woning_2008",
+    "woning_2009",
+    "woning_2010",
+    "woning_2011",
+    "woning_2012",
+    "woning_2013",
+    "woning_2014",
+    "woning_2015",
+    "woning_2016",
+    "woning_2017",
+    "woning_2018",
+    "woning_2019",
+    "woning_2020",
+    "woonvorm_2008",
+    "woonvorm_2009",
+    "woonvorm_2010",
+    "woonvorm_2011",
+    "woonvorm_2012",
+    "woonvorm_2013",
+    "woonvorm_2014",
+    "woonvorm_2015",
+    "woonvorm_2016",
+    "woonvorm_2017",
+    "woonvorm_2018",
+    "woonvorm_2019",
+    "woonvorm_2020",
+    "wave",
+    "nohouse_encr",
+    "positie",
+    "lftdcat",
+    "lftdhhh",
+    "aantalhh",
+    "aantalki",
+    "partner",
+    "burgstat",
+    "woonvorm",
+    "woning",
+    "belbezig",
+    "brutoink",
+    "nettoink",
+    "brutocat",
+    "nettocat",
+    "oplzon",
+    "oplmet",
+    "oplcat",
+    "doetmee",
+    "sted",
+    "simpc",
+    "brutoink_f",
+    "netinc",
+    "nettoink_f",
+    "brutohh_f",
+    "nettohh_f",
+    "werving",
+    "age_imp",
+    "birthyear_imp",
+    "gender_imp",
+    "migration_background_imp",
+]