提交

提交图

3215 次代码提交

作者 SHA1 备注 提交日期
Zihao Ye dfb10db86f [Feature] Bag of samplers for efficient categorical sampling w/ and w/o replacement (#1142)
* upd

* upd

* lint

* upd

* upd

* lint

* upd

* refactor

* lint

* upd

* upd

* upd

* lint

* fix

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* complete test

* upd

* upd

* fix

* vir

* upd

* fix

* fuck

* fix numerical
2020-01-01 11:42:11 +08:00
Da Zheng 162dc19a5d Accelerate edge sampling (#1151)
* accel edge sampling.

* measure sampling time.

* attach positive.

* add neg_deg_sample.

* fix a minor bug.

* remove unnecessary code.

* Revert "add neg_deg_sample."

This reverts commit 45f3c08c87e33589563ca7e6a3ec91cb1239910e.

* Revert "attach positive."

This reverts commit 503d7821555b69b7ccfed55ce8b87ee24fff852d.

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
2019-12-31 01:40:20 -08:00
xiang song(charlie.song) 61b78e6ee7 [Bug Fix]edge sample hotfix (#1152)
* hot fix

* Fix docs

* Fix ArrayHeap float overflow bug

* Fix

* Clean some dead code

* Fix

* FIx

* Add some comments

* run test
2019-12-30 19:42:57 -08:00
Chao Ma 6731ea3af7 [KVStore] Reduce memory cost of kvstore (#1156)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint
2019-12-30 20:25:32 +08:00
Chao Ma 4f02bb751b [KVStore] Made kvstore can support multiple NICs (#1150)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1

* add multiple NICs support for kvstore

* test

* fix lint

* update

* update

* update

* update

* update

* update

* update

* fix lint

* fix lint

* update

* update

* update

* update
2019-12-30 15:17:31 +08:00
Da Zheng f81841532f [KG] reduce memory consumption. (#902)
* reduce memory consumption.

* fix a bug.

* fix a bug.

* fix.
2019-12-28 21:55:50 -08:00
Quan (Andy) Gan 655d75688a fix deprecation warning (#1147) 2019-12-28 12:42:16 +08:00
xiang song(charlie.song) 913e324903 Add device check for sampler input (#1145)
current samplers only support working on CPU
2019-12-27 22:50:52 +08:00
Chao Ma d57ff78da1 [Small Fix] Change default message queue size of communicator to 2GB (#1140)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument

* change default queue size to 2GB

* OMP_NUM_THREADS=1
2019-12-25 17:03:52 +08:00
Minjie Wang 17aab8128a [Doc] update working with multiple backend section (#1128)
* update work with different backend section

* fix some warnings

* Update backend.rst

* Update index.rst

Co-authored-by: VoVAllen <VoVAllen@users.noreply.github.com>
2019-12-24 17:32:56 +08:00
Chao Ma e4ef8d1a80 [KVStore] API change of kvstore (#1058)
* API change of kvstore

* add demo for kvstore

* update

* remove duplicated log

* change queue size

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix lint

* change name

* update

* fix lint

* update

* update

* update

* update

* change message queue size to a python argument
2019-12-24 17:05:15 +08:00
xiang song(charlie.song) 41f8a162f4 [Feature Improvement]Edge Sampler w and w/o replacement (#1116)
* 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: Da Zheng <zhengda1936@gmail.com>
2019-12-24 16:25:40 +08:00
Da Zheng e890a899fc Partition a graph with HALO nodes (#1076)
* get subgraph with halo.

* add partition function.

* add comment.

* parallel partition.

* fix a compilation error.

* fix lint error.

* address comments.

* add comments.

* fix for TF.
2019-12-23 08:46:53 -08:00
VoVAllen 0d9acc9d58 hot fix (#1135) 2019-12-24 00:19:25 +08:00
VoVAllen a1b28b3954 [Hot fix] Fix build threads (#1133)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image

* try node

* try

* fix

* try new ci

* try new ci

* try new ci

* try new ci

* ci

* try

* try

* fix

* hot fix

* fix

* fix cpp test

* add comments
2019-12-23 21:42:59 +08:00
Mufei Li 2ab4d0e10f [Doc] Hotfix for README (#1119) 2019-12-23 19:22:38 +08:00
VoVAllen 7b98e899f3 [CI] Try new CI (#1124)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image

* try node

* try

* fix

* try new ci

* try new ci

* try new ci

* try new ci

* ci

* try

* try

* fix
2019-12-23 17:28:06 +08:00
VoVAllen d30a69bf27 [Backend] TF backend (#978)
* tf

* add builtin support

* fiix

* pytest

* fix

* fix

* fix some bugs

* fix selecting

* fix todo

* fix test

* fix test fail in tf

* fix

* fix

* fix gather row

* fix gather row

* log backend

* fix gather row

* fix gather row

* fix for pytorch

* fix

* fix

* fix

* fix

* fix

* fix tests

* fix

* fix

* fix

* fix

* fix

* fix

* fix convert

* fix

* fix

* fix

* fix inplace

* add alignment setting

* add debug option

* Revert "add alignment setting"

This reverts commit ec63fb3506ea84fff7d447a1fbdfd1d5d1fb6110.

* tf ci

* fix lint

* fix lint

* add tfdlpack

* fix type

* add env

* fix backend

* fix

* fix tests

* remove one_hot

* remove comment

* remove comment

* fix

* use pip to install all

* fix test

* fix base

* fix

* fix

* add skip

* upgrade cmake

* change version

* change ci

* fix

* fix

* fix

* fix

* fix seg fault

* fix

* fix python version

* fix

* try fix

* fix

* fix

* tf takes longer time in ci

* change py version

* fix

* fix

* fix oom

* change kg env

* change kg env

* 啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊啊

* 我再也不搞各种乱七八糟环境了……

* use pytest

* Chang image
2019-12-20 15:56:51 +08:00
Mufei Li cf9ba90fd4 [Chem] ACNN and various utilities (#1117)
* Add several splitting methods

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Finally

* CI
2019-12-19 11:54:40 +08:00
Zihao Ye 6beef85b07 upd (#1113) 2019-12-17 15:24:04 +08:00
John Andrilla 41114b8758 [Doc] Tree-LSTM in DGL, edit pass for readability (#1031)
* [Doc] Tree-LSTM in DGL, edit pass for readability

Edit for grammar and style. 
How about deleting the last line? "Besides, you..." It needs a transition or some context.

* Update tutorials/models/2_small_graph/3_tree-lstm.py

* Update tutorials/models/2_small_graph/3_tree-lstm.py
2019-12-17 14:14:18 +08:00
xiang song(charlie.song) 632a9af8f9 [Feature] No-Uniform Edge Sampler (#1087)
* 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
2019-12-16 11:26:34 -08:00
VoVAllen dd65ee211e [CI] Change tests for flexibility
* change ci image

* fix

* force bash

* fix

* fix python version

* fix

* fix

* fix

* update gpu

* cuda

* jenkins

* fix build sh

* fix

* Revert "fix"

This reverts commit 6b091914b3ef6b4300fa662ee375aa4b27944f5c.

* try fix

* fix

* Revert "fix"

This reverts commit e42c3035fa4974c6b197aaba0748f7b45113ddaa.

* try fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix tests

* try fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix ctx problem

* fix many tests

* fix typo

* add backend

* move to pytorch folder

* fix?

* fix ci

* try skip

* try false

* try?

* try?

* Revert to 7d9a52f

* fix various

* fix lint

* Revert Jenkinsfile

* revert env

* revert env

* address comment

* remove file
2019-12-16 14:54:37 +08:00
John Andrilla f36a451420 [Doc] Old wines new title, edit for grammar and style (#1027)
* Old wines new title, edit for grammar and style

new descriptive title here and an edit pass @aaronmarkham

* Update tutorials/models/4_old_wines/README.txt
2019-12-16 11:24:58 +08:00
naveen marri 2448751c8e Fixed the training file (#1109) 2019-12-14 18:20:05 +08:00
Zihao Ye d16fb3ff9d [Example] GCMC in PyTorch (#1101)
* upd

* fix

* upd

* upd

* upd

* fix

* upd

* upd

* upd

* upd

* upd

* upd

* upd

* upd
2019-12-14 08:18:48 +08:00
Da Zheng 1552090a24 set default backend. (#1104) 2019-12-13 08:30:14 -08:00
Da Zheng adea4ba1d0 fix doc. (#1100) 2019-12-13 05:24:37 -08:00
Da Zheng 869093e890 fix doc of EdgeSampler. (#1093) 2019-12-12 19:15:06 -08:00
Jacob Stevens 4713a33218 Update README.md (#1095)
Extra open brace in the .bibtex entry results in an error
2019-12-13 00:35:57 +08:00
Mufei Li b599f6ed96 Update (#1090) 2019-12-11 13:39:18 +08:00
Quan (Andy) Gan 6ae93e5c6a [Bug] Fix #1088 (#1089)
* [Bug] Fix #1088

* fix

* add comment
2019-12-11 10:27:57 +08:00
Da Zheng 48c7ec4427 [Doc] Fix gcn tutorial (#1080)
* don't have activation.

* add evaluation.

* address comment.
2019-12-10 02:10:51 -08:00
Zihao Ye 870857cfba udp (#1083) 2019-12-09 14:52:20 +08:00
VoVAllen a5a052b5bf [Fix] Fix #1078, #1077 and add nightly build to readme 2019-12-09 00:03:35 +08:00
Zihao Ye fa0ee46a01 [hotfix] node id validity check (#1073)
* fix

* improve

* fix lint

* upd

* fix

* upd
2019-12-06 01:36:41 +08:00
Mufei Li bc4f4352d1 [Chem Model Zoo] Hot fix for README (#1071)
* Fix outdated example

* Hot fix
2019-12-05 14:37:06 +08:00
John Andrilla 7784f0668d [Doc] Small graphs readme.txt, edit pass (#1030)
* [Doc] Small graphs readme.txt, edit pass

Edit for grammar and style. Should this be an .rst on .txt?

* Update README.txt
2019-12-05 13:48:39 +08:00
Guo Qipeng fff3dd9593 [Example] add graphwriter pytorch example (#1068)
* upd

* fig edgebatch edges

* add test

* trigger

* add graphwriter pytorch example

* fix line break in graphwriter README

* upd

* fix
2019-12-05 02:44:15 +08:00
VoVAllen 35653ddd0a [Fix] Fix karate due to networkx version change 2019-12-04 22:02:10 +08:00
John Andrilla 5fc7eb6d97 [Doc] Generative models, edit pass (#1035)
* [Doc] Generative models, edit pass

Edit for grammar and style. Can you clarify "Instead of and/or"?

* Update tutorials/models/3_generative_model/5_dgmg.py

* Update tutorials/models/3_generative_model/5_dgmg.py

Co-Authored-By: Aaron Markham <markhama@amazon.com>

* Update tutorials/models/3_generative_model/5_dgmg.py

Co-Authored-By: Aaron Markham <markhama@amazon.com>

* Update tutorials/models/3_generative_model/5_dgmg.py

Co-Authored-By: Aaron Markham <markhama@amazon.com>

* Update tutorials/models/3_generative_model/5_dgmg.py

Co-Authored-By: Aaron Markham <markhama@amazon.com>
2019-12-03 09:38:47 +08:00
John Andrilla f45178c334 [Doc] Generative models, edit for readability (#1033)
Edit pass for grammar and style
2019-12-02 14:54:10 +08:00
Da Zheng 65a904c565 update instruction. (#1052) 2019-12-01 15:25:42 -08:00
Da Zheng ae743dcc02 remove tqdm. (#1051) 2019-12-01 14:41:45 -08:00
xiang song(charlie.song) dca0e376e0 [KG][Score Func] Update TransE with L2 distance support. (#1059)
* Add L2 distance score for TransE

* Update README.md

* Use linalg.gemm to speedup mx l2 dist

* Fix
2019-12-01 14:03:36 -08:00
John Andrilla d6dfaa9b74 [Doc] Capsule network tutorial, edit pass (#1026)
Edited for grammar and style.
2019-12-01 22:26:51 +08:00
John Andrilla 10b104dd35 [Doc] Transformer tutorial, edit for readability (#1025)
Edit pass for grammar and style
2019-12-01 11:25:51 +08:00
Zihao Ye fdc58a8996 [Hotfix] Fix GAT example and clarify the usage of early stop (#1065)
* upd

* rm redundancy:

* upd
2019-11-30 23:43:15 +08:00
John Andrilla a2b8d8e4e7 [Doc] Giant graph tutorial edit for grammar and style (#1020)
* Edit for readability

* giant_graph_readme edit for grammar

* NodeFlow and Sampling edit pass for grammar

* Update tutorials/models/1_gnn/9_gat.py

* Update tutorials/models/1_gnn/9_gat.py
2019-11-30 12:31:59 +08:00
John Andrilla dd09f15fa7 [Doc] Stochastic steady-state embedding (SSE) Edit for readability
Revisions for grammar and style
2019-11-30 11:37:17 +08:00