def main(model_dir):
np.random.seed(0)
torch.manual_seed(0)
e = Embedding(model_dir)
print(e.nearest_k('god'))
print(e.similarity('god', 'wickedly'))
print(e.analogy('god', 'love', 'satan', k=5))
print()
def main(model_dir):
np.random.seed(0)
torch.manual_seed(0)
e = Embedding(model_dir)
print(e.nearest_k('god'))
print(e.similarity('god', 'wickedly'))
print(e.analogy('god', 'love', 'satan', k=5))
print()
