From d5d2085ea5ce0298e16917ec569e6b1c02c05883 Mon Sep 17 00:00:00 2001 From: Sang Yoon Lee Date: Sat, 28 Nov 2020 12:46:57 +0900 Subject: [PATCH] correct minor error --- src/pre_processing_wine.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/src/pre_processing_wine.py b/src/pre_processing_wine.py index bf0381a..3c8dad5 100644 --- a/src/pre_processing_wine.py +++ b/src/pre_processing_wine.py @@ -12,7 +12,7 @@ --in_file_1= Path (including file name) to first raw data which is for red wine --in_file_2= Path (including file name) to second raw data which is for white wine --out_dir= Path (excluding file name) of where to locally write the file -[--split=] Condsider wheater splict data or not. if --splict ='n', code will split data +[--split=] Condsider wheater splict data or not. if --splict ='n', code will not split data """ from docopt import docopt @@ -44,9 +44,6 @@ def main(in_file_1, in_file_2, out_dir, split): data['quality_rank'] = np.select(conditions, ranks) if split == "n": - # split data as test and train - data_train, data_test = train_test_split(data, test_size=0.2, random_state=123) - # Save data try: data.to_csv(out_dir + "processed.csv", index = False) except: