chavalitros
/
IN-P1
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fixup! Split data using k-fold

This commit is contained in:
coolneng 2020-11-10 20:36:38 +01:00
parent 0421493eff
commit 4652d46966
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/preprocessing.py
View File

@ -31,7 +31,7 @@ def encode_columns(df):
def split_train_target(df): def split_train_target(df):
train_data = df.drop(labels="Severity") train_data = df.drop(columns=["Severity"])
target_data = df["Severity"] target_data = df["Severity"]
return train_data, target_data return train_data, target_data