1. 07 Aug, 2020 1 commit
  2. 09 Jan, 2020 1 commit
  3. 03 Nov, 2019 1 commit
    • Zihao Ye's avatar
      [NN] nn modules & examples update (#890) · 9a0511c8
      Zihao Ye authored
      * upd
      
      * damn it
      
      * fuck
      
      * fuck pylint
      
      * fudge
      
      * remove some comments about MXNet
      
      * upd
      
      * upd
      
      * damn it
      
      * damn it
      
      * fuck
      
      * fuck
      
      * upd
      
      * upd
      
      * pylint bastard
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      9a0511c8
  4. 02 Oct, 2019 1 commit
    • Da Zheng's avatar
      [KG][Model] Knowledge graph embeddings (#888) · 15b951d4
      Da Zheng authored
      * upd
      
      * fig edgebatch edges
      
      * add test
      
      * trigger
      
      * Update README.md for pytorch PinSage example.
      
      Add noting that the PinSage model example under
      example/pytorch/recommendation only work with Python 3.6+
      as its dataset loader depends on stanfordnlp package
      which work only with Python 3.6+.
      
      * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
      
      1. make dgl.nn.xxx frame agnostic
      2. make test.backend include dgl.nn modules
      3. modify test_edge_softmax of test/mxnet/test_nn.py and
          test/pytorch/test_nn.py work on both CPU and GPU
      
      * Fix style
      
      * Delete unused code
      
      * Make agnostic test only related to tests/backend
      
      1. clear all agnostic related code in dgl.nn
      2. make test_graph_conv agnostic to cpu/gpu
      
      * Fix code style
      
      * fix
      
      * doc
      
      * Make all test code under tests.mxnet/pytorch.test_nn.py
      work on both CPU and GPU.
      
      * Fix syntex
      
      * Remove rand
      
      * Add TAGCN nn.module and example
      
      * Now tagcn can run on CPU.
      
      * Add unitest for TGConv
      
      * Fix style
      
      * For pubmed dataset, using --lr=0.005 can achieve better acc
      
      * Fix style
      
      * Fix some descriptions
      
      * trigger
      
      * Fix doc
      
      * Add nn.TGConv and example
      
      * Fix bug
      
      * Update data in mxnet.tagcn test acc.
      
      * Fix some comments and code
      
      * delete useless code
      
      * Fix namming
      
      * Fix bug
      
      * Fix bug
      
      * Add test for mxnet TAGCov
      
      * Add test code for mxnet TAGCov
      
      * Update some docs
      
      * Fix some code
      
      * Update docs dgl.nn.mxnet
      
      * Update weight init
      
      * Fix
      
      * init version.
      
      * change default value of regularization.
      
      * avoid specifying adversarial_temperature
      
      * use default eval_interval.
      
      * remove original model.
      
      * remove optimizer.
      
      * set default value of num_proc
      
      * set default value of log_interval.
      
      * don't need to set neg_sample_size_valid.
      
      * remove unused code.
      
      * use uni_weight by default.
      
      * unify model.
      
      * rename model.
      
      * remove unnecessary data sampler.
      
      * remove the code for checkpoint.
      
      * fix eval.
      
      * raise exception in invalid arguments.
      
      * remove RowAdagrad.
      
      * remove unsupported score function for now.
      
      * Fix bugs of kg
      Update README
      
      * Update Readme for mxnet distmult
      
      * Update README.md
      
      * Update README.md
      
      * revert changes on dmlc
      
      * add tests.
      
      * update CI.
      
      * add tests script.
      
      * reorder tests in CI.
      
      * measure performance.
      
      * add results on wn18
      
      * remove some code.
      
      * rename the training script.
      
      * new results on TransE.
      
      * remove --train.
      
      * add format.
      
      * fix.
      
      * use EdgeSubgraph.
      
      * create PBGNegEdgeSubgraph to simplify the code.
      
      * fix test
      
      * fix CI.
      
      * run nose for unit tests.
      
      * remove unused code in dataset.
      
      * change argument to save embeddings.
      
      * test training and eval scripts in CI.
      
      * check Pytorch version.
      
      * fix a minor problem in config.
      
      * fix a minor bug.
      
      * fix readme.
      
      * Update README.md
      
      * Update README.md
      
      * Update README.md
      15b951d4
  5. 08 Sep, 2019 1 commit
    • xiang song(charlie.song)'s avatar
      [Kernel][Perf] Message builtin with broadcasting performance optimization (#815) · bcd33e0a
      xiang song(charlie.song) authored
      * upd
      
      * fig edgebatch edges
      
      * add test
      
      * trigger
      
      * Update README.md for pytorch PinSage example.
      
      Add noting that the PinSage model example under
      example/pytorch/recommendation only work with Python 3.6+
      as its dataset loader depends on stanfordnlp package
      which work only with Python 3.6+.
      
      * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
      
      1. make dgl.nn.xxx frame agnostic
      2. make test.backend include dgl.nn modules
      3. modify test_edge_softmax of test/mxnet/test_nn.py and
          test/pytorch/test_nn.py work on both CPU and GPU
      
      * Fix style
      
      * Delete unused code
      
      * Make agnostic test only related to tests/backend
      
      1. clear all agnostic related code in dgl.nn
      2. make test_graph_conv agnostic to cpu/gpu
      
      * Fix code style
      
      * fix
      
      * doc
      
      * Make all test code under tests.mxnet/pytorch.test_nn.py
      work on both CPU and GPU.
      
      * Fix syntex
      
      * Remove rand
      
      * Add TAGCN nn.module and example
      
      * Now tagcn can run on CPU.
      
      * Add unitest for TGConv
      
      * Fix style
      
      * For pubmed dataset, using --lr=0.005 can achieve better acc
      
      * Fix style
      
      * Fix some descriptions
      
      * Test performance of udf
      
      * trigger
      
      * Fix doc
      
      * Add nn.TGConv and example
      
      * Update test code
      
      * Fix bug
      
      * Update data in mxnet.tagcn test acc.
      
      * Fix some comments and code
      
      * delete useless code
      
      * Fix namming
      
      * Fix bug
      
      * Fix bug
      
      * Add test for mxnet TAGCov
      
      * Add test code for mxnet TAGCov
      
      * Update some docs
      
      * Fix some code
      
      * Update docs dgl.nn.mxnet
      
      * Update weight init
      
      * Fix
      
      * Minor opt for URRevel
      
      * Delete test code
      
      * Update code style and notes.
      
      * Fix func name
      bcd33e0a
  6. 30 Aug, 2019 1 commit
  7. 28 Aug, 2019 1 commit
    • xiang song(charlie.song)'s avatar
      [NN] Add MXNet impl for TAGCN module. (#799) · e17add56
      xiang song(charlie.song) authored
      * upd
      
      * fig edgebatch edges
      
      * add test
      
      * trigger
      
      * Update README.md for pytorch PinSage example.
      
      Add noting that the PinSage model example under
      example/pytorch/recommendation only work with Python 3.6+
      as its dataset loader depends on stanfordnlp package
      which work only with Python 3.6+.
      
      * Provid a frame agnostic API to test nn modules on both CPU and CUDA side.
      
      1. make dgl.nn.xxx frame agnostic
      2. make test.backend include dgl.nn modules
      3. modify test_edge_softmax of test/mxnet/test_nn.py and
          test/pytorch/test_nn.py work on both CPU and GPU
      
      * Fix style
      
      * Delete unused code
      
      * Make agnostic test only related to tests/backend
      
      1. clear all agnostic related code in dgl.nn
      2. make test_graph_conv agnostic to cpu/gpu
      
      * Fix code style
      
      * fix
      
      * doc
      
      * Make all test code under tests.mxnet/pytorch.test_nn.py
      work on both CPU and GPU.
      
      * Fix syntex
      
      * Remove rand
      
      * Add TAGCN nn.module and example
      
      * Now tagcn can run on CPU.
      
      * Add unitest for TGConv
      
      * Fix style
      
      * For pubmed dataset, using --lr=0.005 can achieve better acc
      
      * Fix style
      
      * Fix some descriptions
      
      * trigger
      
      * Fix doc
      
      * Add nn.TGConv and example
      
      * Fix bug
      
      * Update data in mxnet.tagcn test acc.
      
      * Fix some comments and code
      
      * delete useless code
      
      * Fix namming
      
      * Fix bug
      
      * Fix bug
      
      * Add test code for mxnet TAGCov
      
      * Update some docs
      
      * Fix some code
      
      * Update docs dgl.nn.mxnet
      
      * Update weight init
      
      * Fix
      e17add56
  8. 23 Aug, 2019 1 commit
    • Minjie Wang's avatar
      [NN] RGCN modules (#744) · 708765f0
      Minjie Wang authored
      * 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
      708765f0
  9. 06 Aug, 2019 1 commit
  10. 05 Aug, 2019 1 commit
    • Zihao Ye's avatar
      [Feature] DGL Pooling modules (#669) · 5d3f470b
      Zihao Ye authored
      * removal doc
      
      * glob
      
      * upd
      
      * rm knn
      
      * add softmax
      
      * upd
      
      * upd
      
      * add broadcast and s2s
      
      * optimize max_on
      
      * forsaken changes to heterograph
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * bugfix
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * format upd
      
      * upd format
      
      * upd doc
      
      * upd
      
      * import order
      
      * upd
      
      * rm warnings
      
      * fix
      
      * upd test
      
      * upd
      
      * upd
      
      * fix device
      
      * upd
      
      * upd
      
      * upd
      
      * upd
      
      * remove 1.1
      
      * upd
      
      * trigger
      
      * trigger
      
      * add more tests
      
      * fix device
      
      * upd
      
      * upd
      
      * refactor
      
      * fix?
      
      * fix
      
      * upd docstring
      
      * refactor
      
      * upd
      
      * fix
      
      * upd
      
      * upd
      
      * upd
      
      * fix
      
      * upd docs
      
      * add shape
      
      * refactor & upd doc
      
      * upd doc
      
      * upd
      5d3f470b
  11. 25 Feb, 2019 1 commit
    • Minjie Wang's avatar
      [WIP] [NN] Refactor NN package (#406) · 565f0c88
      Minjie Wang authored
      * refactor graph conv
      
      * docs & tests
      
      * fix lint
      
      * fix lint
      
      * fix lint
      
      * fix lint script
      
      * fix lint
      
      * Update
      
      * Style fix
      
      * Fix style
      
      * Fix style
      
      * Fix gpu case
      
      * Fix for gpu case
      
      * Hotfix edgesoftmax docs
      
      * Handle repeated features
      
      * Add docstring
      
      * Set default arguments
      
      * Remove dropout from nn.conv
      
      * Fix
      
      * add util fn for renaming
      
      * revert gcn_spmv.py
      
      * mx folder
      
      * fix wierd bug
      
      * fix mx
      
      * fix lint
      565f0c88