This commit is contained in:
jakubknczny 2021-05-16 16:04:31 +02:00
parent c815688698
commit 407d6187af

View File

@ -7,13 +7,16 @@ import tensorflow
from tensorflow.keras import layers from tensorflow.keras import layers
ex = Experiment("470607", interactive=False, save_git_info=False) ex = Experiment("470607", interactive=False, save_git_info=False)
ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred')) ex.observers.append(
MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred'))
ex.observers.append(FileStorageObserver('my_runs')) ex.observers.append(FileStorageObserver('my_runs'))
@ex.config @ex.config
def my_config(): def my_config():
learning_rate = float(sys.argv[1]) learning_rate = float(sys.argv[1])
@ex.capture @ex.capture
def prepare_train_model(learning_rate, _run): def prepare_train_model(learning_rate, _run):
_run.info["prepare_model"] = str(datetime.now()) _run.info["prepare_model"] = str(datetime.now())
@ -28,10 +31,10 @@ def prepare_train_model(learning_rate, _run):
Y_valid = pd.get_dummies(Y_valid) Y_valid = pd.get_dummies(Y_valid)
model = tensorflow.keras.Sequential([ model = tensorflow.keras.Sequential([
layers.Input(shape=(12,)), layers.Input(shape=(12,)),
layers.Dense(32), layers.Dense(32),
layers.Dense(16), layers.Dense(16),
layers.Dense(2, activation='softmax') layers.Dense(2, activation='softmax')
]) ])
model.compile( model.compile(
@ -43,7 +46,8 @@ def prepare_train_model(learning_rate, _run):
model.save('grid-stability-dense.h5') model.save('grid-stability-dense.h5')
_run['history'] = str(history) _run['history'] = str(history.history[:, -1])
@ex.main @ex.main
def my_main(learning_rate): def my_main(learning_rate):
@ -52,4 +56,3 @@ def my_main(learning_rate):
r = ex.run() r = ex.run()
ex.add_artifact('grid-stability-dense.h5') ex.add_artifact('grid-stability-dense.h5')