Skip to main content

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


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)

Uploaded Source

Built Distribution

kgeditor-1.0.0-py3-none-any.whl (41.5 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page