s495719/ium_495719
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

try

Browse Source
This commit is contained in:
PawelDopierala 2024-04-02 23:18:46 +02:00
parent e77989bf86
commit d19f741dbf
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data_processing.py
View File

@ -1,10 +1,11 @@
import sklearn import sklearn
from sklearn.model_selection import train_test_split
import pandas as pd import pandas as pd
housing_price_dataset = pd.read_csv('housing_price_dataset.csv') housing_price_dataset = pd.read_csv('housing_price_dataset.csv')
hp_train_test, hp_dev = sklearn.model_selection.train_test_split(housing_price_dataset, test_size=0.1) hp_train_test, hp_dev = train_test_split(housing_price_dataset, test_size=0.1)
hp_train, hp_test = sklearn.model_selection.train_test_split(hp_train_test, test_size=1000) hp_train, hp_test = train_test_split(hp_train_test, test_size=1000)
hp_train = pd.get_dummies(hp_train, columns=['Neighborhood']) hp_train = pd.get_dummies(hp_train, columns=['Neighborhood'])
hp_dev = pd.get_dummies(hp_dev, columns=['Neighborhood']) hp_dev = pd.get_dummies(hp_dev, columns=['Neighborhood'])