Fix trainY and testY

sebastianstock · Jun 12, 2017 · 3065d99 · 3065d99
1 parent 90a2946
commit 3065d99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/intro-to-tflearn/TFLearn_Sentiment_Analysis_Solution.ipynb b/intro-to-tflearn/TFLearn_Sentiment_Analysis_Solution.ipynb
@@ -319,8 +319,8 @@
     "test_fraction = 0.9\n",
     "\n",
     "train_split, test_split = shuffle[:int(records*test_fraction)], shuffle[int(records*test_fraction):]\n",
-    "trainX, trainY = word_vectors[train_split,:], to_categorical(Y.values[train_split], 2)\n",
-    "testX, testY = word_vectors[test_split,:], to_categorical(Y.values[test_split], 2)"
+    "trainX, trainY = word_vectors[train_split,:], to_categorical(Y.values[train_split, 0], 2)\n",
+    "testX, testY = word_vectors[test_split,:], to_categorical(Y.values[test_split, 0], 2)"
    ]
   },
   {