|
|
@@ -26,7 +26,7 @@ def rmse(y_true, y_pred):
|
|
|
return sqrt
|
|
|
|
|
|
|
|
|
-var_dir = os.path.dirname(os.path.dirname(__file__))
|
|
|
+# var_dir = os.path.dirname(os.path.dirname(__file__))
|
|
|
var_dir = './j00645-w-xch'
|
|
|
|
|
|
|
|
|
@@ -100,7 +100,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=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