kdudzic/ium_s449288
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'simple_regression.py'
All checks were successful
s449288-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Kacper Dudzic 2022-04-24 18:05:57 +02:00
parent 4664e19d2e
commit b061491605
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
simple_regression.py
View File

@ -4,6 +4,10 @@ from keras.models import save_model
import pandas as pd import pandas as pd
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import sys
# Pobranie przykładowego argumentu trenowania
EPOCHS_NUM = int(sys.argv[1])
# Wczytanie danych # Wczytanie danych
data_train = pd.read_csv('lego_sets_clean_train.csv') data_train = pd.read_csv('lego_sets_clean_train.csv')
@ -35,7 +39,7 @@ model.compile(
history = model.fit( history = model.fit(
train_piece_counts, train_piece_counts,
train_prices, train_prices,
epochs=100, epochs=EPOCHS_NUM,
verbose=0, verbose=0,
validation_split=0.2 validation_split=0.2
) )