alex-tifrea / poincare_glove

Implementation of the "Poincare Glove: Hyperbolic word embeddings" paper
GNU Lesser General Public License v2.1
83 stars 20 forks source link

Update poincare.py #14

Open bonaventuredossou opened 3 years ago

bonaventuredossou commented 3 years ago

added encoding = "utf-8" for csv reader for the Link Prediction and Reconstruction file reading modified find_matching_terms() to bring best possibilities in the model's vocabulary closed to the input word, instead of considering only words starting by the input words, leading to "KeyError" error