It is the implementation we used for our experiments.
It is not the final version. We will clean up our code and make it more easier to use.
- Download Wizard of Wikipedia Dataset and ConceptNet 5.5
- Run Process.py to process the raw data.
- Run Datasets.py, that will create a cache file (Remember the directory!)
- Use the implementation of ConvE to get pre-trained weights.
- Run main.py. If you want to change settings, check
settings/train_config.json
、settings/eval_config.json
andsettings/model_config.json
- More details can be found in our source code. (But it may be hard to read. Sorry for the bad coding style :( )