nju-websoft / OpenEA

A Benchmarking Study of Embedding-based Entity Alignment for Knowledge Graphs, VLDB 2020
GNU General Public License v3.0
517 stars 80 forks source link

Add prediction method for given entities #19

Closed sven-h closed 3 years ago

sven-h commented 3 years ago

Hi,

I added the function to predict the confidence values of given entities. Furthermore I changed the parameter output_path to output_file_name of the predict function to only specify the file name and not the full path. The file will be written in the out folder.

Additionally I find it quite useful to have the embeddings in a textual format. I choose here the word2vec C text format which is also used by gensim. This allows to read the embeddings easily in other programming languages. If you think that this is too much during saving, we can also make it optional.

As usual, if you have any thoughts on how to make it better, I would be happy to hear it.

Best regards Sven