mirrors / dmlc--dgl

文件历史

3,215 提交 34 分支 42 Git 标签
fix-gpu-sampling-doc
提交图

3 次代码提交

作者 SHA1 备注 提交日期
Hongzhi (Steve), Chen a9f2acf323 [Misc] Black auto fix. (#4641) 
* [Misc] Black auto fix.

* sort

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-09-26 21:46:32 +08:00
Minjie Wang 708765f0a1 [NN] RGCN modules (#744) 
* rgcn module

* support id input

* WIP: model codes

* use faster index select

* dropout

* self loop

* WIP: link prediction

* fix lint

* WIP: docs

* docstring

* docstring

* merge two child classes

* mxnet rgcn module

* fix lint

* fix lint

* fix rename bug

* add uniform edge sampler

* fix fn name

* docstring

* fix mxnet rgcn module

* fix mx rgcn

* enable test on cuda
 2019-08-23 16:38:48 -04:00
Yizhi Liu 2c5b48ab60 [Model][MXNet] RGCN Entity Classification (#246) 
* entity classify work for examples

* add loop_msg

* remove wrong assert

* remove one reshape

* add readme

* add MRR

* remove mrr from entity task
 2018-12-05 10:27:59 -08:00