chore: merge branch 'main' of github.com:ninpnin/probabilistic-word-embeddings

Author: ninpnin

README.md

@@ -1,4 +1,4 @@
-# probabilistic-word-embeddings v1.13.7
+# probabilistic-word-embeddings v1.16.0
 
 Probabilistic Word Embedding module for Python. Built with TensorFlow 2.x and TensorFlow probability.
 

docs/probabilistic_word_embeddings.html

@@ -200,9 +200,9 @@ def words_in_e(row):
     r = len(df)
     target_words = list(df[columns[-1]])
 
-    X1 = embedding[df[columns[0]]]
-    X2 = embedding[df[columns[1]]]
-    X3 = embedding[df[columns[2]]]
+    X1 = embedding[list(df[columns[0]])]
+    X2 = embedding[list(df[columns[1]])]
+    X3 = embedding[list(df[columns[2]])]
     X = X1 - X2 + X3
 
     inv_vocab = {v: k for k, v in e.vocabulary.items()}