From fde81b35a4c1845af7da41f7c95eaa793254d279 Mon Sep 17 00:00:00 2001 From: Zeynep Akkalyoncu Yilmaz Date: Tue, 18 Dec 2018 19:52:29 -0500 Subject: [PATCH] Add type checking for np arrays : --- lib/data/util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/data/util.py b/lib/data/util.py index 1c8f151..8e53f9f 100644 --- a/lib/data/util.py +++ b/lib/data/util.py @@ -99,6 +99,7 @@ def reverse_index(indexed_data, vocab, ravel=False): reversed_data = list() indexed_data = np.argmax(indexed_data, axis=-1) word_idx = {id: word for word, id in vocab.items()} + if isinstance(indexed_data, np.ndarray): indexed_data = indexed_data.tolist() for indexed_line in indexed_data: if ravel: reversed_data.append([' '.join((word_idx[x] for x in indexed_line[1:len(indexed_line)-1]))])