intercepter fix + heuristics

decryptoai/decryptoai/players/unsupervised/naive_heuristics.py

@@ -0,0 +1,17 @@
+import numpy as np
+
+def cosine_similarity(embedding, clue_index, keyword_index):
+    clue_embedding = embedding[clue_index.squeeze()].transpose()
+    keyword_embedding = embedding[keyword_index.squeeze()]
+    return np.expand_dims(keyword_embedding.dot(clue_embedding), axis=-1)
+
+# simple heuristics
+
+def log_square_cosine_similarity(embedding, clue_index, keyword_index):
+    similarity = cosine_similarity(embedding, clue_index, keyword_index)
+    return 2 * np.log(np.abs(similarity))
+
+def log_normalized_cosine_similarity(embedding, clue_index, keyword_index):
+    similarity = cosine_similarity(embedding, clue_index, keyword_index)
+    normalized_similiarity = (1 + similarity) / 2
+    return np.log(normalized_similiarity)

decryptoai/decryptoai/players/unsupervised/numpy_intercepter.py

@@ -1,5 +1,5 @@
 import numpy as np
-import numpy_guesser as npg
+import decryptoai.players.unsupervised.numpy_guesser as npg
 from functools import partial
 # shape-preserving normalization