s434695/ium_434695
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix train
Some checks failed
s434695-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
s434695 2021-05-16 21:50:54 +02:00
parent ef245b69da
commit 26ddaa3a0c
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

@ -26,7 +26,7 @@ def my_config():
batch_param = int(sys.argv[1]) batch_param = int(sys.argv[1])
epoch_param = int(sys.argv[2]) epoch_param = int(sys.argv[2])
def regression_model(epoch_param, batch_param, _run): def prepare_model(epoch_param, batch_param, _run):
_run.info["prepare_model_ts"] = str(datetime.now()) _run.info["prepare_model_ts"] = str(datetime.now())
# odczytanie danych z plików # odczytanie danych z plików
vgsales_train = pd.read_csv('train.csv') vgsales_train = pd.read_csv('train.csv')
@ -66,7 +66,7 @@ def regression_model(epoch_param, batch_param, _run):
@ex.main @ex.main
def my_main(epoch_param, batch_param): def my_main(epoch_param, batch_param):
print(regression_model()) print(prepare_model())
r = ex.run() r = ex.run()
ex.add_artifact("vgsales_model.h5") ex.add_artifact("vgsales_model.h5")