added some argparse arguments to main

2017-07-03 13:48:12 +02:00
parent c972963a19
commit 5f8a760a0c
2 changed files with 63 additions and 8 deletions
--- a/models.py
+++ b/models.py
@@ -45,8 +45,8 @@ def get_top_cnn(cnn, numFeatures, maxLen, windowSize, domainFeatures, filters, k
    maxPool = GlobalMaxPooling1D()(cnn)
    cnnDropout = Dropout(cnnDropout)(maxPool)
    cnnDense = Dense(cnnHiddenDims, activation='relu')(cnnDropout)
-    cnnOutput1 = Dense(2, activation='softmax')(cnnDense)
-    cnnOutput2 = Dense(2, activation='softmax')(cnnDense)
+    cnnOutput1 = Dense(2, activation='softmax', name="client")(cnnDense)
+    cnnOutput2 = Dense(2, activation='softmax', name="server")(cnnDense)

    # We define a trainable model linking the
    # tweet inputs to the predictions