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.model_selection import train_test_split
|
||||||
from sklearn.preprocessing import MinMaxScaler
|
from sklearn.preprocessing import MinMaxScaler
|
||||||
from datasets import load_dataset
|
from datasets import load_dataset
|
||||||
@ -20,3 +21,7 @@ val[numerical_features] = scaler.fit_transform(val[numerical_features])
|
|||||||
train.dropna(inplace=True)
|
train.dropna(inplace=True)
|
||||||
test.dropna(inplace=True)
|
test.dropna(inplace=True)
|
||||||
val.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