We use two typical datasets FB15K-237 (Toutanova et al., 2015) and NELL-995 (Xiong et al., 2017) for training and evaluation.
we set K = 137 and K = 114 to select few-shot relations from FB15K-237 and NELL-995 respectively.