Repository files navigation
Developing completed (Prototype)
Require pytorch>=1.6.0 for CPU (torch.sparse.softmax)
Require pytorch>=1.7.0 for GPU (CUDA torch.sparse.softmax)
fix bugs in val and test dataset and dataloader
add kinship dataset
add k-hop option for dataset
val and test running on cpu
enable batch training
enable batch to val and test also
add customized collate function to dataloader for different length batch input
modify filter to filter all other triples
remove for loop in load data and clean up
add None to attention type
change ranking data from scipy to torch
add reverse function
add GAT for models.py
add ConvE and separate DistMult for score_function.py
add early stopping
ConvAttenion code polished
add GPU support
add label smoothing
add filter for model results
change name to GINN
ranking part code polished
change loss to BCEloss as ConvE (1-N scoring)
remove negative sampling
add new ops_new dataset for extensive test triple
add WN18RR dataset
change measurement from ACC to MRR
clean layers.py and parameterized
create parser.py
add negative sampling
create utils.py
initial commit
add README
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
You can’t perform that action at this time.