From de8d36d32d1355abae6794c2b7f917c64b823b8d Mon Sep 17 00:00:00 2001 From: zehua zhang Date: Tue, 23 Jul 2019 17:22:20 -0400 Subject: [PATCH] Update generate_stylometric.py If run this file, there will be error: TypeError("Parameter doc_words of infer_vector() must be a list of strings (not a single string)."). So need to change data[i][1] to a list of string. --- users/generate_stylometric.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/users/generate_stylometric.py b/users/generate_stylometric.py index 1befc1d..0e658fd 100755 --- a/users/generate_stylometric.py +++ b/users/generate_stylometric.py @@ -15,7 +15,7 @@ wr = csv.writer(file, quoting=csv.QUOTE_ALL) # Inferring paragraphVec vectors for each user -vectors = np.asarray([doc2vec.infer_vector(data[i][1]) for i in range(data.shape[0])]) +vectors = np.asarray([doc2vec.infer_vector([data[i][1]]) for i in range(data.shape[0])]) users = data[:,0] for i in range(len(users)):