new dataset format: multi-lists -> two arrays

2017-07-04 09:18:50 +02:00
parent 5f8a760a0c
commit 5743127b7f
2 changed files with 65 additions and 44 deletions
--- a/main.py
+++ b/main.py
@@ -104,10 +104,6 @@ def main():
    client_labels = client_labels[idx]
    server_labels = server_tr[idx]

-    # TODO: remove when features are flattened
-    for i in range(len(X_tr)):
-        X_tr[i] = X_tr[i][idx]
-
    shared_cnn = models.get_shared_cnn(len(char_dict) + 1, embeddingSize, maxLen,
                                       domainFeatures, kernel_size, domainFeatures, 0.5)