|
|
@@ -116,7 +116,7 @@ class FMI(object):
|
|
|
# print("weight_lstm_1 = ", weight_lstm_1)
|
|
|
# print("bias_lstm_1 = ", bias_lstm_1)
|
|
|
|
|
|
- check_point = ModelCheckpoint(filepath='./var/' + 'fmi.h5', monitor='val_loss',
|
|
|
+ check_point = ModelCheckpoint(filepath=os.path.join(var_dir, 'var', 'fmi.h5'), monitor='val_loss',
|
|
|
save_best_only=True, mode='auto')
|
|
|
early_stop = EarlyStopping(monitor='val_loss', patience=opt.Model['patience'], mode='auto')
|
|
|
# tbCallBack = TensorBoard(log_dir='../figure',
|
David