LARS-research / AutoSF

Y. Zhang, Q. Yao, J. Kwok. Bilinear Scoring Function Search for Knowledge Graph Learning. TPAMI 2022
68 stars 12 forks source link

How to get the triplet classification result? #8

Open ideafang opened 2 years ago

ideafang commented 2 years ago

Hi, I'm very interested in this meaningful work. And I notice that the triplet classification result is shown in the paper "AutoSF: Searching Scoring Functions for Knowledge Graph Embedding". But it does not seem to exist in this repository.

yzhangee commented 2 years ago

Hi, thanks for your interest. The implementation of triplet classification task is quite easy. You can refer to the code in OpenKE https://github.com/thunlp/OpenKE for testing. The training process is same as our code here.