From f675f8017deefdd715c8ea07bcf9935e523ee707 Mon Sep 17 00:00:00 2001 From: Natalia Szymczyk Date: Fri, 21 Apr 2023 10:50:31 +0200 Subject: [PATCH] fix python script --- main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/main.py b/main.py index 471d4f8..2719af5 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,4 @@ +import pandas as pd from sklearn.model_selection import train_test_split from sklearn.preprocessing import MinMaxScaler from datasets import load_dataset @@ -20,3 +21,7 @@ val[numerical_features] = scaler.fit_transform(val[numerical_features]) train.dropna(inplace=True) test.dropna(inplace=True) val.dropna(inplace=True) + +train.to_csv('liver.train.data') +test.to_csv('liver.test.data') +val.to_csv('liver.dev.data')