search

allenai / commonsense-kg-completion

MIT License
107 stars 16 forks source link

About the pre-trained BERT ConceptNet embedding #3

Closed zsun227 closed 4 years ago

zsun227 commented 4 years ago

Hello! Thanks for sharing your code and pre-trained embeddings.

I wonder how can we know the corresponding word for every bert embedding vector in the file "conceptnet_bert_embeddings.pt"? Looks like it's shape is 78334 x 1024. But the "cn_node_names.txt" contains 78249 entities,,,,, So I'm not sure how to link each entity to its pre-trained BERT embedding. Could you give me some insights? thanks!

chaitanyamalaviya commented 4 years ago

Hi! The provide embeddings include the precomputed embeddings for the evaluation nodes as well. Here is a list of the 78334 node names which can be linked to the provided embeddings in the same order.

zsun227 commented 4 years ago

Hi! The provide embeddings include the precomputed embeddings for the evaluation nodes as well. Here is a list of the 78334 node names which can be linked to the provided embeddings in the same order.

Thank you!