Update process_data.py

src/data/process_data.py

@@ -13,6 +13,7 @@ def process_data(split="train"):
     df = df.sample(frac=params["split"], replace=True, random_state=1)
     if os.path.exists("data/raw/{}.csv".format(split)):
         os.remove("data/raw/{}.csv".format(split))
+    df.to_csv("data/processed/{}.csv".format(split))
 
 
 if __name__ == '__main__':