added params

2017-07-07 16:48:10 +02:00
parent 3c4be52bb6
commit be56112b33
4 changed files with 191 additions and 72 deletions
--- a/models/pauls_networks.py
+++ b/models/pauls_networks.py
@@ -2,6 +2,20 @@ import keras
 from keras.engine import Input, Model
 from keras.layers import Embedding, Conv1D, GlobalMaxPooling1D, Dense, Dropout, Activation, TimeDistributed

+best_config = {
+    'domain_features': 32,
+    'drop_out': 0.5,
+    'embedding_size': 64,
+    'filter_main': 512,
+    'flow_features': 3,
+    'hidden_dims': 32,
+    'filter_embedding': 32,
+    'hidden_embedding': 32,
+    'kernel_embedding': 8,
+    'kernels_main': 8,
+    'input_length': 40
+}
+

 def get_embedding(vocab_size, embedding_size, input_length,
                  filters, kernel_size, hidden_dims, drop_out=0.5):