文件历史

8 次代码提交

作者 SHA1 备注 提交日期
blokhinnv 731660d23e fixed typos in docs and docstring (#3231)
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-08-20 02:41:21 +00:00
zhjwy9343 2fa7f71ad5 [Doc] new nn api doc (#2019)
* Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [Feature] Add dotproduct attention

* [New] Update landing page

* [New] Update landing page

* [New] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Doc] Update landing page

* [Improvement] use dgl build-in in dotgatconv

* [Doc] review API doc string bottom up

* [Doc] Add doc of input and output features

* [Doc] Update doc string for pooling and transformer.

* [Doc] Reformat doc string and change some wordings.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

* [Doc] Doc string refactoring.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2020-08-14 18:00:38 +08:00
Minjie Wang f25bc176d0 [Hetero] Improve speed of several Hetero APIs (#1486)
* add clone function to frame

* add utest

* replace all local_var with local_scope

* fix utest

* avoid creating canonical types in __getitem__

* lint

* try another utest  appraoch for mx

* utest
2020-05-01 15:21:33 +08:00
Minjie Wang 3efb5d8ecf [NN] Add HeteroGraphConv module for cleaner module definition (#1385)
* Add HeteroGraphConv

* add custom aggregator; some docstring

* debugging

* rm print

* fix some acc bugs

* fix initialization problem in weight basis

* passed tests

* lint

* fix graphconv flag; add error message

* add mxnet heteroconv

* more fix for mx

* lint

* fix torch cuda test

* fix mx test_nn

* add exhaust test for graphconv

* add tf heteroconv

* fix comment
2020-03-27 18:47:06 +08:00
Zihao Ye aa06e4579f [Doc] Improve the docstring of several nn modules. (#1187)
* upd

* upd

* upd

* upd

* lint

* upd

* upd

* upd

* upd

* upd

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-09 21:14:33 +08:00
Zihao Ye ca302a13ac [Feature] Add dgl.nn.*.Sequential for usability (#1166)
* upd

* upd

* upd

* upd

* lint

* upd

* upd

* upd

* upd

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2020-01-09 12:39:22 +08:00
VoVAllen ddb5d804f9 [Refactor] Break NN modules into files (#859)
* break nn modules into files

* break mxnet nn modules

* fix lint

* fix lint
2019-09-16 14:44:15 +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