explicit type conversion

zylamarek · Apr 4, 2017 · 742c6dd · 742c6dd
1 parent f3d438c
commit 742c6dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluation_helper.py b/evaluation_helper.py
@@ -118,7 +118,7 @@ def __evaluate_dataset(self, data_id):
             acc += np.sum(np.equal(predicted_labels, y_batch) * mask_tweet)
 
             # create confusion matrix for current batch
-            mask_cm = (1 - mask_tweet) * self.data.n_labels * self.data.n_labels
+            mask_cm = (np.ones_like(mask_tweet) - mask_tweet) * self.data.n_labels * self.data.n_labels
             labels, label_counts = np.unique(y_batch * self.data.n_labels + predicted_labels + mask_cm,
                                              return_counts=True)
             for i, c in enumerate(labels):