add parser argument for naming in multi model modes, minor fixes,

This commit is contained in:
2017-09-05 12:40:37 +02:00
parent 1da31cc97c
commit ed4f478bad
6 changed files with 34 additions and 20 deletions

View File

@@ -75,7 +75,7 @@ def get_new_model(dropout, flow_features, domain_features, window_size, domain_l
y = Conv1D(cnn_dims,
kernel_size,
activation='relu',
input_shape=(window_size, domain_features + flow_features))(merged)
input_shape=(window_size, domain_features + flow_features))(y)
# remove temporal dimension by global max pooling
y = GlobalMaxPooling1D()(y)
y = Dropout(dropout)(y)