A library that provides a tool to edit model easily.
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
kgeditor-1.0.0.tar.gz
(38.7 kB
view hashes)
Built Distribution
kgeditor-1.0.0-py3-none-any.whl
(41.5 kB
view hashes)