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

Update 'train.py'

Browse Source
This commit is contained in:
Tomasz Koszarek 2023-09-30 01:00:00 +02:00
parent e7dd668578
commit 5cfd930e4f
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -25,9 +25,10 @@ def main(EPOCHS, _run):
if EPOCHS == 0:
EPOCHS = 500
train_data_x = pd.read_csv('./train_data.csv')
train_data_x_tmp = train_data_x.copy()
price_train = train_data_x.copy()
price_predict = train_data_x.pop('Production_year')
price_predict = train_data_x_tmp.pop('Index')
normalize = layers.Normalization()
normalize.adapt(price_train)