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 00:56:22 +02:00
parent 1b6c86630f
commit e7dd668578
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

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