From 11768a9460b231eb45e639443128add251ada49e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Wojciech=20Miko=C5=82ajski?= Date: Sat, 17 Jun 2023 20:24:05 +0200 Subject: [PATCH] Update 'train.py' --- train.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index d549a49..0bf4350 100644 --- a/train.py +++ b/train.py @@ -1,15 +1,14 @@ import pandas as pd import tensorflow as tf from sklearn.preprocessing import StandardScaler, OneHotEncoder +from sklearn.model_selection import train_test_split +from sklearn import preprocessing water = pd.read_csv('waterQuality1.csv') water = water[water['is_safe'].apply(lambda x: str(x).isdigit())] water['is_safe'].value_counts() - -from sklearn.model_selection import train_test_split -from sklearn import preprocessing water_min_max = preprocessing.MinMaxScaler() water_min_max = water_min_max.fit_transform(water) water_min_max = pd.DataFrame(water_min_max, columns=water.columns)