python/dglke/models/mxnet/score_fun.py
python/dglke/models/pytorch/score_fun.py