fix python script

This commit is contained in:
Natalia Szymczyk 2023-04-21 10:50:31 +02:00
parent 99bb6baaa8
commit f675f8017d
1 changed files with 5 additions and 0 deletions

View File

@ -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')