文件历史

提交图

243 次代码提交

作者 SHA1 备注 提交日期
Mufei Li e9c3c0e8ad [Model] Simplify RGCN
* Update (#5)

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* FIx

* Try

* Update

* Update

* Update

* Fix

* Update

* Fix

* Fix

* Fix

* Fix

* Update

* Fix

* Update

* Update

* Update

* Fix

* Fix

* Update

* Update

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Update

* Update

* Update README.md

* Update

* Fix

* Update

* Update

* Fix

* Fix

* Fix

* Update

* Update

* Update

Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal>

* Update

* Update

* Fix

* Update

* Update

* Update

* Fix

* Update

* Update

* Update

* Update

* Update

* Update

* CI

Co-authored-by: Ubuntu <ubuntu@ip-172-31-6-240.us-west-2.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-57-123.us-west-2.compute.internal>
Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2022-02-17 15:33:27 +08:00
RhettYing 6106a99d0a refine 2022-02-17 02:00:10 +00:00
RhettYing e0f054fb27 [Doc] add doc for DGLCSVDataset 2022-02-17 00:52:16 +00:00
Rhett Ying 39121dfdb8 [Feature] support non-numeric node_id/src_id/dst_id/graph_id and rena… (#3740)
* [Feature] support non-numeric node_id/src_id/dst_id/graph_id and rename CSVDataset

* change return value when iterate dataset

* refine data_parser

* force reload
2022-02-17 08:35:39 +08:00
Mufei Li 8b8fd2c0be [Dataset] Add transform argument to built-in datasets (#3733)
* Update

* Fix

* Update
2022-02-15 16:45:47 +08:00
Jinjing Zhou dc78e11c3f [Dataset] Adapter to convert a dataset for link prediction task (#3699)
* add ut

* add doc link

* install dep

* add

* Revert "install dep"

This reverts commit e574a8377144749056c6849b655004df2771e179.

* add

* merge fix

* rm files

* fix

* fix

* fix

* fix

* fix typo

* fix tf

* fix

* fix

* fix

* fix

* fix

* fix dependency

* fix test

* fix

* fix

* add doc

* fix

* fix

* fix test

* fix test

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2022-02-11 16:59:14 +08:00
Rhett Ying 45ac57260a [data] refine AsNodePredDataset and add tests for DGLCSVDataset (#3722)
* [data] refine AsNodePredDataset and add tests for DGLCSVDataset

* fix

* remove add_self_loop

* refine
2022-02-10 17:46:39 +08:00
Minjie Wang 5152a87982 [Data] Utility function and class for converting a dataset for node prediction (#3695)
* add ut

* add doc link

* install dep

* fix  ci

* fix ut; more comments

* remove deprecated attributes in rdf datasets; fix label feature name

* address comments

* fix ut for other frameworks
2022-01-30 14:31:37 +08:00
Rhett Ying 9ec9df57ca [Fix] check and load dependencies when needed (#3655)
* [Fix] check and load dependencies when needed

* refine rdflib import
2022-01-18 16:10:47 +08:00
Rhett Ying 41124b3284 [Fix] Do not import DGLCSVDataset in default in case of dependency issue (#3642) 2022-01-11 15:25:03 +08:00
Rhett Ying 95c0ff6302 [Feature] add CSVDataset to load data from csv files (#3547)
* [Feature] add CSVDataset to load data from csv files

* add CSVDataset class file

* install pyyaml when running unit tests

* install pandas for unit tests

* utilize pydantic for YAML config check

* generate yaml and csv files during test

* make more keys as optional

* remove/rename several keys in yaml config and more tets though looks a bit clumsy

* fix test failure on mxnet

* pass /path/to/dataset instead of yaml path

* code refinement

* code refine

* change several yaml field such as feat and graph_id

* merge graph generation logic

* refine code

* Refactored_first_version

* DGLCSVDataset works for single heterograph

* add more tests

* fix test failure in mxnet

* add docstring

* use list comprehension for dict

* fix version in YAML

* refine data length assert

* use dict.pop directly

* remove ambiguous variable names

* refine graph id missing logic

* refine graph create call

* separate node/edge/graph data parser

* remove separator in DefaultDataParser

* refine validation error log for yaml field

* minor check

* refine code via dict.get()

* move load_from_csv into Node/Edge/GraphData

* move _parse_node/edge/graph_data into Node/Edge/GraphData

* refine id-related fields check

* check duplicate ntypes/etypes when load yaml

* refine docstring
2022-01-10 17:21:57 +08:00
esang a7b5085a5d Add a note about the order of TUDataset (#3549) 2021-11-28 00:12:29 +08:00
Yuchen 7c771d0d6c [BugFix] fix #3429 and update results of caregnn (#3441)
* squeeze node labels in FraudDataset

* fix RLModule

* update results in README.md

* fix KeyError in full graph training

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-11-10 16:00:12 +08:00
Rhett Ying f9fd7fd7f7 [BugFix] extract gz into target dir (#3389) 2021-09-30 11:35:40 +08:00
Kay Liu fecd6f3c50 [BugFix] fix typo in fakenews dataset variable name (#3363) 2021-09-21 22:06:26 +08:00
Mufei Li 8c2f8ff84a Update citation_graph.py (#3221) 2021-08-06 15:57:42 +08:00
Yuchen 4ae13bd270 [bugfix] Fix force_reload parameter of FraudDataset (#3210)
* enable force_reload of FraudDataset

* rewrite hash_key in FraudDataset
2021-08-03 13:14:17 +08:00
Kay Liu a107993f10 [Model] add model example CARE-GNN (#3187)
* [Model] add model example CARE-GNN

* update README

* improvements based on the review feedback

* fix missing item()

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-07-29 15:24:15 +08:00
Xiangkun Hu bbb6d4eed0 [Dataset] Fix the edge numbers in the GNNBenchmarkDataset doc string (#3163)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets

* add four papers

* Fix doc string

* fix readme

* fix readme

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-07-21 17:31:32 +08:00
Quan (Andy) Gan 066bef21f0 [Doc] Fix CoraFull number of edges inconsistency in doc (#3139) 2021-07-14 15:06:57 +08:00
Rhett Ying 186ef59283 [Feature] apply dgl.reorder() onto several node classification datase… (#3102)
* [Feature] apply dgl.reorder() onto several node classification datasets in DGL

* rebase on latest dgl.reorder_graph()
2021-07-13 08:52:17 +08:00
Kay Liu 0ce92a86ed [BugFix] fix problems found in bug bash (#3116)
* fix breakline in fakenews.py

* fix inconsistent argument name

* modify incorrect example and deprecated graph type

* modify docstring and example in knn_graph

* fix incorrect node type

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
2021-07-12 13:34:50 +08:00
Kay Liu bb89dee7ff [BugFix] fix problems in data split (#3082)
* [BugFix] fix problems in data split

* fix format problems in docstring

* modify statistics to fit in dgl nature

Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-07-07 16:07:23 +08:00
Quan (Andy) Gan 7823ae2dbb add TUDataset notice (#3075)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-06-30 14:22:45 +08:00
Quan (Andy) Gan acd21a6d60 [Feature] Support direct creation from CSR and CSC (#3045)
* csr and csc creation

* fix

* fix

* fixes to adj transpose

* fine

* raise error if indptr did not match number of nodes

* fix

* huh?

* oh

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-06-25 10:05:47 +08:00
Mufei Li ff519f98c3 [API] Standardize Subgraph APIs (#2929)
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Fix subgraph tests

* Capture stdout for distributed test

* Capture stdout for distributed test

* Update

* Update

* Update

* Update subgraph.cc

Co-authored-by: Ubuntu <ubuntu@ip-172-31-28-17.us-west-2.compute.internal>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-06-21 19:53:37 +08:00
Kay Liu 9706eaa895 [Feature] add permission information and fix import problems (#3036)
* [Feature] add positive negative statistics

* [Feature] add permission information and fix import problem

* fix backend incompatible problem

* modify random split to remove sklearn usage

* modify file read to remove pandas usage

* add datasets into doc

* add random seed in data splitting

* add dataset unit test

* usage permission information update

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-06-21 11:55:15 +08:00
zhjwy9343 55e7796ac5 Revert "[Feature] add permission information and fix import problems (#2998)" (#3029)
This reverts commit cba5b188b5.

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-06-17 13:58:38 +08:00
Kay Liu cba5b188b5 [Feature] add permission information and fix import problems (#2998)
* [Feature] add positive negative statistics

* [Feature] add permission information and fix import problem

* fix backend incompatible problem

* modify random split to remove sklearn usage

* modify file read to remove pandas usage

* add datasets into doc

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-06-16 18:24:17 +08:00
Leonardo Castorina 63aed58204 [Doc] Fix typo in DGL base dataset class (#2977) #3017
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-06-15 16:15:52 +08:00
Jinjing Zhou a303f07819 fix #2952 (#3010) 2021-06-14 15:09:14 +08:00
Kay Liu 849cbec6ee [Feature] add positive negative statistics (#2984)
Co-authored-by: zhjwy9343 <6593865@qq.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-06-05 15:48:11 +08:00
Kay Liu b8b0fd22b2 [Feature] add two fraud datasets (#2908)
* add two fraud dataset

* add comments and modify to heterogenous graph

* modify the node name based on datasets

* add assertion message

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-05-28 15:45:44 +08:00
Kay Liu 64d2a2aff7 [Feature] fix #2876, add two fake news datasets (#2939)
* [Feature] fix #2876, add two fake news datasets

* add name assertion

* add assertion message

Co-authored-by: zhjwy9343 <6593865@qq.com>
2021-05-28 14:29:26 +08:00
Quan (Andy) Gan 1fe0860750 [Bugfix] Fixes #2844 (#2847)
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-25 18:12:17 +08:00
Xiangkun Hu b49cd0952f [Bugfix] Fix feature type of 'ENZYMES' in TUDataset (#2800)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets

* add four papers

* fix feature type of TU dataset

* fix readme

* remove commented codes

Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-04-01 14:36:55 +08:00
milesial 1f4c0b7193 Fix processing of QM9EdgeDataset (#2801) 2021-04-01 10:13:51 +08:00
Hengrui Zhang bcffdb82c9 [Example] Add implementation of mvgrl (#2739)
* [Example ]add mvgrl

* [Doc] add mvgrl to readme

* add more comments

* fix typos

* replace tab with space

* [doc] replace tab with space

* [Doc] fix a typo

* fix minor typos

* fix typos

* fix typos

* fix typos

* fix typos

* fix

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-22 21:20:21 +08:00
xnouhz 0b47e86803 [Example] Add DimeNet(++) for Molecular Graph Property Prediction (#2706)
* [example] arma

* update

* update

* update

* update

* update

* [example] dimenet

* [docs] update dimenet

* [docs] update tf results

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

* update

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-09 17:08:59 +08:00
Hengrui Zhang a1f59c3b68 [Feature] QM9Edge Dataset Support (#2704)
* [Feature] Support QM9Edge Datset

* Update qm9_edge.py

* disable tqdm

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* Update qm9_edge.py

* remove preprocessing part

* add comparisons in qm9.py

* [docs] add qm9edge dataset

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-03-09 14:51:21 +08:00
Tong He 7298173ac1 [Doc]Improve docstring for GINDataset (#2676)
* add docstring for gin

* update doc

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
2021-02-23 15:14:10 +08:00
Mufei Li b1840f49fa [Dataset] Change the Data Type of the Node Features in GINDataset/TUDataset from Float64 to Float32 (#2592)
* Update

* Update

* Update

* update

Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
Co-authored-by: Tong He <hetong007@gmail.com>
2021-02-03 21:00:45 +08:00
Yu Sun 20d49771df [Feature] Add reverse edges in CitationGraphDataset (#2588)
* [Feature] Add reverse edges in CitationGraphDataset

* fix bug

* fix bug

* fix bug

* Update python/dgl/data/citation_graph.py

Co-authored-by: Xiangkun Hu <huxk_hit@qq.com>

* fix notes

* fix notes

* fix bugs

* solve requested changes

* fix bug

* fix bug

Co-authored-by: Xiangkun Hu <huxk_hit@qq.com>
2021-02-02 15:52:43 +08:00
Tong He 12f6429616 remove hash_func from dataset object (#2572)
Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
2021-01-27 18:49:40 +08:00
Jinjing Zhou c45f6eb575 [Bugfix] Fix duplicate edges in Coauthor Dataset #2553 (#2569)
* fix

* address comment
2021-01-26 15:30:27 +08:00
Da Zheng 25ac334403 [Distributed] Heterogeneous graph support (#2457)
* Distributed heterograph (#3)

* heterogeneous graph partition.

* fix graph partition book for heterograph.

* load heterograph partitions.

* update DistGraphServer to support heterograph.

* make DistGraph runnable for heterograph.

* partition a graph and store parts with homogeneous graph structure.

* update DistGraph server&client to use homogeneous graph.

* shuffle node Ids based on node types.

* load mag in heterograph.

* fix per-node-type mapping.

* balance node types.

* fix for homogeneous graph

* store etype for now.

* fix data name.

* fix a bug in example.

* add profiler in rgcn.

* heterogeneous RGCN.

* map homogeneous node ids to hetero node ids.

* fix graph partition book.

* fix DistGraph.

* shuffle eids.

* verify eids and their mappings when loading a partition.

* Id map from homogneous Ids to per-type Ids.

* verify partitioned results.

* add test for distributed sampler.

* add mapping from per-type Ids to homogeneous Ids.

* update example.

* fix DistGraph.

* Revert "add profiler in rgcn."

This reverts commit 36daaed8b660933dac8f61a39faec3da2467d676.

* add tests for homogeneous graphs.

* fix a bug.

* fix test.

* fix for one partition.

* fix for standalone training and evaluation.

* small fix.

* fix two bugs.

* initialize projection matrix.

* small fix on RGCN.

* Fix rgcn performance (#17)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix lint.

* fix.

* fix test.

* fix lint.

* test partitions.

* remove redundant test for partitioning.

* remove commented code.

* fix partition.

* fix tests.

* fix RGCN.

* fix test.

* fix test.

* fix test.

* fix.

* fix a bug.

* update dmlc-core.

* fix.

* fix rgcn.

* update readme.

* add comments.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>

* fix.

* fix.

* add div_int.

* fix.

* fix.

* fix lint.

* fix.

* fix.

* fix.

* adjust.

* move code.

* handle heterograph.

* return pytorch tensor in GPB.

* remove some tests in example.

* add to_block for distributed training.

* use distributed to_block.

* remove unnecessary function in DistGraph.

* remove distributed to_block.

* use pytorch tensor.

* fix a bug in ntypes and etypes.

* enable norm.

* make the data loader compatible with the old format.

* fix.

* add comments.

* fix a bug.

* add test for heterograph.

* support partition without reshuffle.

* add test.

* support partition without reshuffle.

* fix.

* add test.

* fix bugs.

* fix lint.

* fix dataset.

* fix for mxnet.

* update docstring.

* rename to floor_div

* avoid exposing NodePartitionPolicy and EdgePartitionPolicy.

* fix docstring.

* fix error.

* fixes.

* fix comments.

* rename.

* rename.

* explain IdMap.

* fix docstring.

* fix docstring.

* update docstring.

* remove the code of returning heterograph.

* remove argument.

* fix example.

* make GraphPartitionBook an abstract class.

* fix.

* fix.

* fix a bug.

* fix a bug in example

* fix a bug

* reverse heterograph sampling.

* temp fix.

* fix lint.

* Revert "temp fix."

This reverts commit c450717b9f578b8c48769c675f2a19d6c1e64381.

* compute norm.

* Revert "reverse heterograph sampling."

This reverts commit bd6deb7f52998de76508f800441ff518e2fadcb9.

* fix.

* move id_map.py

* remove check

* add more comments.

* update docstring.

Co-authored-by: Ubuntu <ubuntu@ip-172-31-2-202.us-west-1.compute.internal>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-9-132.us-west-1.compute.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-62-171.ec2.internal>
2021-01-24 19:18:00 -08:00
xnouhz 2c6d071617 [Feature] QM9 Dataset Support (#2521)
* [Model] MixHop for Node Classification task

* [docs] update

* [docs] update

* [fix] remove seed option

* [fix] update readme

* [feature] support qm9 dataset

* [style] update

* [docs] fix the details

* [fix] indexing only support int

* [style] update

* [fix] multiple backends support

* [docs] add qm9

* [fix] Z type: float32 -> int32

* [fix] Z type: int32 -> long

* [docs] add ref

* [docs] fix

* [docs] update

* [docs] update

* [fix] test eval

* [docs] fix example

Co-authored-by: xnuohz@126.com <ubuntu@ip-172-31-44-184.us-east-2.compute.internal>
Co-authored-by: Mufei Li <mufeili1996@gmail.com>
Co-authored-by: Jinjing Zhou <VoVAllen@users.noreply.github.com>
2021-01-21 01:33:30 +08:00
张天启 9a9a06eb2f [Bug Fix] Fix undefined var bug in LegacyTUDataset (#2543)
* [Bug Fix] Fix undefined var bug in LegacyTUDataset

* upt

* [Bug Fix] Fix cache file name bug in TUDataset

Co-authored-by: Tong He <hetong007@gmail.com>
2021-01-19 13:58:01 +08:00
Xiangkun Hu 21e90cbf48 [Doc] Fix number of classes in Amazon Cobuy datasets (#2488)
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Update gnn_benckmark.py

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets
2021-01-05 13:08:05 +08:00
xiang song(charlie.song) 4464b9734c Fix kg dataset len() (#2475)
Co-authored-by: Ubuntu <ubuntu@ip-172-31-59-204.ec2.internal>
2020-12-31 23:33:21 +08:00