KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#dsh #dsh-plugin #ai-agent #developer-tools #codex #typescript #cli #deepseek-harness #claude-code #ai-agents #mcp #rust

DIFT

★ 35 Open GitHub ↗

Finetuning Generative Large Language Models with Discrimination Instructions for Knowledge Graph Completion, ISWC 2024

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# DIFT
Finetuning Generative Large Language Models with Discrimination Instructions for Knowledge Graph Completion, ISWC 2024

## requirements
    pytorch==2.1.0
    bitsandbytes==0.40.0
    transformers==4.31.0
    peft==0.4.0
    accelerate==0.21.0
    einops==0.6.1
    evaluate==0.4.0
    scikit-learn==1.2.2
    sentencepiece==0.1.99
    wandb==0.15.3

## Dataset
download from [here](https://drive.google.com/file/d/1gWX97jtILkf960f_zBbkLoIRpO54Cv7E/view?usp=drive_link), extract the files, and move them to the corresponding folders.

## Finetuning
### FB15K237
    bash script/train_fb.sh {TransE|SimKGC|CoLE}

### WN18RR
    bash script/train_wn.sh {TransE|SimKGC|CoLE}

## Inference
    bash script/eval.sh {FB15K237|WN18RR} {TransE|SimKGC|CoLE} {checkpoint_dir}

`checkpoint_dir` is the path of the folder to save the PEFT model, like "./output/FB15K237/2024xxxx-xxxxxx/checkpoint-xxxx/adapter_model"

[checkpoints](https://drive.google.com/file/d/1YH6PUUl81i9gHpboK9SQAHw3zVOuVgzI/view?usp=drive_link) of the reported results are also provided.

DIFT

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README