s487174/ium_z487174
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'dataset-create.py'

Browse Source
This commit is contained in:
Tomasz Koszarek 2023-09-30 00:33:49 +02:00
parent 04af7163a0
commit d9f0fa862a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dataset-create.py
View File

@ -22,8 +22,9 @@ if csv_file is None:
data = pd.read_csv(csv_file)
data.sample(CUTOFF)
data_tmp = data.copy()
df = pd.DataFrame(data.pop('Price'))
df = pd.DataFrame(data_tmp.pop('Price'))
train_data, temp_data, train_data2, temp_data2 = train_test_split(data, df, test_size=0.4, random_state=42)
dev_data, test_data, dev_data2, test_data2 = train_test_split(temp_data, temp_data2, test_size=0.4, random_state=42)