fix python script
This commit is contained in:
parent
99bb6baaa8
commit
f675f8017d
5
main.py
5
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')
|
||||
|
Loading…
Reference in New Issue
Block a user