Skip to content

Netralab/custom-ogbl-split

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

custom-ogbl-split

Create split for link property prediction task for your custom datasets.

Step 1: Run the demo

python3 pos_neg_split.py --dataset Drug-Drug.edgelist --splitting_strategy random

Step 2: Unpack/Unzip the generated files and get the split_dict.dt for getting the splitted edges

python get_the_sets.py

About

Create split for link property prediction task for your custom datasets

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages