A Library for Dynamically Editing PLMs-Based Knowledge Graph Embeddings.
Project description
KGEditor
Code for our paper “Editing Language Model-based Knowledge Graph Embeddings”.
To use as a python module
pip install -kgeditor
Run the experiments
Training
Edit Task
To train the KGEditor model in the paper on the dataset E-FB15k237, run the command below. For another dataset E-WN18RR just replacing the dataset name will be fine.
./scripts/kgeditor_fb15k237_edit.sh
Add Task
To train the KGEditor model in the paper on the dataset A-FB15k237, run the command below. For another dataset A-WN18RR just replacing the dataset name will be fine.
./scripts/kgeditor_fb15k237_add.sh
Models
This folder contains the base models used for this work.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
DeltaKG-0.0.5.tar.gz
(68.7 kB
view hashes)
Built Distribution
DeltaKG-0.0.5-py3-none-any.whl
(71.9 kB
view hashes)