Overall, in this project, we first consider multiple types of embeddings such as ComplEx, Transe, Abstract, Lexicalized embeddings and so on. Then, we use different sets of basis datasets for retrofitting these embeddings (bringing the neighbouring nodes closer to each other). We then evaluate the original and retrofitted embeddings on various evaluation benchmark datasets.
All the server files are uploaded on Google Drive here
Most of the results mentioned in the paper are derived from csv sheets generated from the Retrofitting notebook. The final versions of these results sheets can be found on Google Drive here
Other files used can be found here:
- Embedding Datasets - Ignore the files in the folders
- Wikidata Child-Parent and Sibling Datasets
- Evaluation Benchmark Datasets
In this notebook, we generate embeddings using ComplEx, Transe, lexicalized properties, abstract, H, A, S, labels only, labels + descriptions, and concatenated embeddings. In this notebook, we also generate direct similarity scores for pairs of nodes in evaluation benchmark datasets based on class, JC, topSim for direct evaluation in the Retrofitting notebook.
In the basis notebook, we generate the basis datasets using 2 relations - parent-child and siblings which we will later use in the Retrofitting notebook. In the probase notebook, we generate another such basis dataset.
In this notebook, we generate the evaluation benchmark datasets on which we evaluate our embeddings performance based on similarity both before and after retrofitting wherever applicable. We generate benchmarks for Wordsim353 (Annotated), DBPedia sourced datasets (MC 30, RG 65), ConceptNet, Wiki-CS but go on to using Wordsim and DBPedia ones only in the Retrofitting notebook. In the ConcepNet Evaluation Dataset Exploration notebook, we just explore the ConceptNet dataset downloaded but the main file generation for our evaluation purpose is done in the main Evaluation Datasets notebook.
This is the main framework code where all the retrofitting, evaluation is done and results are captured.
Datasets Used:
- Source ConceptNet File
- Source Wiki-CS File
- Source Probase Dataset
- DBPedia Abstracts Dataset
- Wikidata DWD v2
- Sitelinks from Wikidata 2020-12-08
Project Draw.IO files stored on Google Drive here