1. 05 Jan, 2020 1 commit
    • Da Zheng's avatar
      [KG] PBG's way of constructing negative edges (#1159) · 1022d5d6
      Da Zheng authored
      * attach positive.
      
      * add neg_deg_sample.
      
      * add comment.
      
      * add neg_deg_sample for eval.
      
      * change the edge sampler.
      
      * rename edge sampler in KG.
      
      * allow specifying chunk size and negative sample size separately.
      
      * fix bugs in KG.
      
      * add check in sampler.
      
      * add more checks.
      
      * fix
      
      * add comment.
      
      * add comments.
      1022d5d6
  2. 04 Jan, 2020 1 commit
  3. 03 Jan, 2020 1 commit
  4. 29 Dec, 2019 1 commit
  5. 24 Dec, 2019 1 commit
    • xiang song(charlie.song)'s avatar
      [Feature Improvement]Edge Sampler w and w/o replacement (#1116) · 41f8a162
      xiang song(charlie.song) authored
      
      
      * Add weight based edge sampler
      
      * Can run, edge weight work.
      TODO: test node weight
      
      * Fix node weight sample
      
      * Fix y
      
      * Update doc
      
      * Fix syntex
      
      * Fix
      
      * Fix GPU test for sampler
      
      * Fix test
      
      * Fix
      
      * Refactor EdgeSampler to act as class object not function that it
      can record its own private states.
      
      * clean
      
      * Fix
      
      * Fix
      
      * Fix run bug on kg app
      
      * update
      
      * update test
      
      * test
      
      * Simply python API and fix some C code
      
      * Fix
      
      * Fix
      
      * Fix syntex
      
      * Fix
      
      * Update API description
      
      * add replacement for edge sampler
      
      * Now edge sampler support replacement and no-replacement
      
      * Fix
      
      * Fix
      
      * change kg/app to use edge sampler with replacement config
      
      * Update replacement algo
      
      * Fix syntax
      
      * Update
      
      * Update
      Co-authored-by: default avatarDa Zheng <zhengda1936@gmail.com>
      41f8a162
  6. 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