From 4652d469662a3bd47c41d9cb6dbc63b4291d3adc Mon Sep 17 00:00:00 2001
From: coolneng <akasroua@gmail.com>
Date: Tue, 10 Nov 2020 20:36:38 +0100
Subject: [PATCH] fixup! Split data using k-fold

---
 src/preprocessing.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/preprocessing.py b/src/preprocessing.py
index 0dcc9bf..c6335a8 100644
--- a/src/preprocessing.py
+++ b/src/preprocessing.py
@@ -31,7 +31,7 @@ def encode_columns(df):
 
 
 def split_train_target(df):
-    train_data = df.drop(labels="Severity")
+    train_data = df.drop(columns=["Severity"])
     target_data = df["Severity"]
     return train_data, target_data