s434743/ium_z434743
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'training.py'

Browse Source
This commit is contained in:
Patryk Łukasiewicz 2023-05-12 19:22:18 +02:00
parent 25c1ef1b9d
commit cf674db27c
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
training.py
View File

@ -9,10 +9,12 @@ import random
exint = Experiment('z-s434743-training')
exint.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017',
db_name='sacred'))
exint.observers.append(FileStorageObserver('z-s434743-training/master'))
@exint.config
def my_config():
EPOCHS = int(os.environ['EPOCHS'])
@ -20,7 +22,7 @@ def my_config():
@exint.main
def my_main(EPOCHS, _run):
_run.info["epochs"] = EPOCHS,
_run.info["epochs"] = EPOCHS
normalize = layers.Normalization()
train_data_x = pandas.read_csv('./X_train.csv')
_run.info["data"] = train_data_x
@ -46,5 +48,7 @@ def my_main(EPOCHS, _run):
exint.run()
exint.add_artifact('test/saved_model.pb')
exint.add_source_file('./ium_z434743/train.py')