slicing
This commit is contained in:
parent
c815688698
commit
407d6187af
@ -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')
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user