mirrors / dmlc--dgl

文件历史

2,937 提交 34 分支 42 Git 标签
test_sparse_compile
提交图

47 次代码提交

作者 SHA1 备注 提交日期
Guangyu Zhou 178290249e [Dataset] Add PATTERN dataset (#5422) 
* add PATTERN dataset

* fix bug

* fix bugs

* fix issues

* refine according to dongyu's comments

---------

Co-authored-by: BuptTab <gyzhou2000@gmail.com>
Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-03-08 14:08:12 +08:00
Zhiteng Li a53ecd229b [Dataset] Add CLUSTER dataset (#5389) 
* add CLUSTER dataset

* refine according to dongyu's comments

---------

Co-authored-by: rudongyu <ru_dongyu@outlook.com>
 2023-03-01 16:36:12 +08:00
Hongzhi (Steve), Chen a208e8868b [Misc] Black auto fix. (#4680) 
* [Misc] Black auto fix.

* fix pylint disable

Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
 2022-10-08 14:15:56 +08:00
RecLusIve-F defa292bc0 [Dataset] Add Flickr and Yelp dataset (#4099) 
* Add Flickr and Yelp dataset

* Update flickr.py

* update

* Update yelp.py

* Update yelp.py

* update

* Update yelp.py

* Update test_data.py

* Update yelp.py

* update

* Update test_data.py

* Update yelp.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-06-14 16:14:12 +08:00
RecLusIve-F 89655cfda2 [Dataset] Add WikiCS Dataset (#4035) 
* Fix bugs & Update dataset

* Update

* Update wikics.py

* Update wikics.py

* Update test_data.py

* Update wikics.py

* Update wikics.py

* Update wikics.py

* update

* Update module.py

* Update dgl.data.rst

* Update wikics.py

* Update wikics.py

Co-authored-by: Mufei Li <mufeili1996@gmail.com>
 2022-06-02 23:03:17 +08:00
Mufei Li d9c25521bc [Data] AsGraphPredDataset (#4073) 
* Update

* CI

* Update

* Update

* Fix

* Fix
 2022-06-02 16:18:48 +08:00
Mufei Li 65b0b9e8c3 [Dataset & Transform] Synthetic Datasets for Explainability and SIGNDiffusion Transform (#3982) 
* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Update

* Fix

* Update

* Update

* Update
 2022-05-16 22:32:09 +08: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
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
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
Quan (Andy) Gan 066bef21f0 [Doc] Fix CoraFull number of edges inconsistency in doc (#3139) 2021-07-14 15:06:57 +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
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
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
Minjie Wang f13b9b6212 [Doc] Scan the API docs and make many changes (#2080) 
* WIP: api

* dgl.sampling, dgl.data

* dgl.sampling; dgl.dataloading

* sampling packages

* convert

* subgraph

* deprecate

* subgraph APIs

* All docstrings for convert/subgraph/transform

* almost all funcs under dgl namespace

* WIP: DGLGraph

* done graph query

* message passing functions

* lint

* fix merge error

* fix test

* lint

* fix

Co-authored-by: Quan Gan <coin2028@hotmail.com>
 2020-08-20 23:26:49 +08:00
Tong He 967ecb8064 [Dataset] Fix the docstring format for dgl.data section (#1941) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* update data rst

* update data doc and docstring

* API doc rst for dataset

* docstring

* update api doc

* add url format

* update docstring

* update citation graph

* update knowledge graph

* update gc datasets

* fix index

* Rst fix (#3)

* Fix syntax

* syntax

* update docstring

* update doc (#4)

* final update

* fix rdflib

* fix rdf

Co-authored-by: HuXiangkun <huxk_hit@qq.com>
Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-06 14:47:49 +08:00
Xiangkun Hu 451ed6d836 [Dataset] SBMMixtureDataset (#1920) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* SBMMixture dataset

* Update sbm.py

* fix example

* Update sbm.py

* Revert "Update sbm.py"

This reverts commit 066db5c89bd5e159981ae7cad1bfb883ea5db71d.

* Update sbm.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-04 15:06:53 +08:00
Xiangkun Hu 37aa99c510 [Dataset] GNNBenchmarkDataset (#1912) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* gnn benchmark dataset

* Update gnn_benckmark.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-03 23:28:58 +08:00
Xiangkun Hu 001d793711 [Dataset] SSTDataset (#1918) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* SSTDataset

* Update tree.py
 2020-08-03 19:15:28 +08:00
Xiangkun Hu 73b9c6f18c [Dataset] GDELTDataset (#1911) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* gdelt dataset

* Update gdelt.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-03 17:43:50 +08:00
Xiangkun Hu 06ea03d0f8 [Dataset] ICEWS18Dataset (#1913) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* ICEWS18Dataset

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-03 17:07:02 +08:00
Xiangkun Hu 22485e58d3 [Dataset] BitcoinOTCDataset (#1910) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* bitcoinotc dataset

* Update bitcoinotc.py
 2020-08-03 12:15:04 +08:00
Xiangkun Hu d22f96feb4 [Dataset] QM7bDataset (#1915) 
* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* QM7bDataset

* Update qm7b.py

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-03 11:58:31 +08:00
Xiangkun Hu 743c38401c [Dataset] KarateClubDataset (#1897) 
* karateclub dataset

* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Update karate.py

* doc string

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>
 2020-08-02 22:39:11 +08:00
xiang song(charlie.song) b347590a37 [Dataset] Citation graph (#1902) 
* citation graph

* GCN example use new citatoin dataset

* mxnet gat

* triger

* Fix

* Fix gat

* fix

* Fix tensorflow dgi

* Fix appnp, graphsage for mxnet

* fix monet and sgc for mxnet

* Fix tagcn

* update sgc, appnp

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>
 2020-08-01 19:25:50 +08:00
Mufei Li 90d2118da5 [Dataset] Change Cora Split (#1583) 
* Use new version of Cora

* Fix import
 2020-06-03 15:45:38 +08:00
VoVAllen 9c790b1143 [Dataset] Add CoraFull, Amazon, KarateClub, Coauthor Dataset (#855) 
* convert np.ndarray to backend tensor

* add datasets

* add qm7

* add dataset

* add dataset

* fix

* change ppi

* tu dataset

* add datasets

* fix

* fix

* fix

* fix

* add docstring

* docs

* doc
 2019-10-01 01:19:59 +08:00
Mufei Li e590feeb62 [Model Zoo] GAT on Tox21 (#793) 
* GAT

* Fix mistake

* Fix

* hotfix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Fix

* Update

* Update

* Update

* Fix style

* Hotfix

* Hotfix

* Hotfix

* Fix

* Fix

* Update

* CI trial

* Update

* Update

* Update
 2019-08-28 04:47:16 +08:00
lunar 3bc73931f8 [Model Zoo] Molecule Regression (#779) 
* [Model] MPNN

* [Model] MPNN 🔨 reorganize the mpnn/sch/mgcn model & alchemy dataset

* [Model] MPNN alchemy dataloader refactoring

* [Model] Chem model zoo minor change

* [Model] Chem Model Zoo 🔥 remove old samples

* [Model Zoo] molecule regression minor change

* Fix dataset import

* Fix dataset import

* [Model Zoo] molecule regression test set

* [Model Zoo] molecule prediction MPNN model hyperparameter tuning

* [Model Zoo] molecule prediction ✏️ mpnn performance update
 2019-08-26 17:42:33 +08:00
VoVAllen 2ee3c78c7c [Dataset] Tox21 (#760) 
* tox21

* fix ci

* fix ci

* fix urls to url

* add doc

* remove binary
 2019-08-14 16:55:01 +08:00
Da Zheng 8c79885d8b [BUGFIX] fix the error message in the data loader. (#509) 
* fix.

* Change to required=True

* Update description

* Change back
 2019-04-26 09:52:45 +09:00
kitaev-chen a3febc061b [Model] Add GIN Model (#471) 
* add gin model

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module

* convert dataset.py to data_ont_the_fly way and put it into dgl.data module
python code checked

* modified document and reference TUDataset; checked python part and bypass cpp part due to error

* change tensor to numpy in dataset and transform in collate@Dataloader

* Change minor format issue

Change minor format issue

* moved logging; adjusted tqdm etc
 2019-04-17 14:57:07 +08:00
VoVAllen 00fc680aae [Dataset] Add TUDataset (#473) 
* add graph classification dataset

* add node label

* add TUDataset

* Modify to consistent with Qi Huang's implementation

* add docs

* Add docs

* Fix change of environment variable

* Update tu.py

* Update tu.py

* Fix error when add node with np.int64
 2019-04-10 09:37:53 -04:00
Hao Zhang 788d8dd4ad [Model]PPI dataloader and inductive learning script. (#395) 
* Create ppi.py

* Create train_ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Create gat.py

* Update train.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* Update train_ppi.py

* Update ppi.py

* Update train_ppi.py

* update docs and readme
 2019-02-17 16:41:30 -05:00
Ziyue Huang 51293a3273 [Data] reddit data loader (#372) 
* reddit data loader

* upload to S3

* update

* add self loop

* address comments

* fi
 2019-02-08 10:53:39 -08:00
Minjie Wang abf12fc757 [Data] Synthetic dataset for graph classificaiton (#364) 
* minigc dataset

* more comments

* sphinx
 2019-01-25 10:47:47 -05:00
HQ b1e8d95e99 [Doc] LGNN tutorial (#210) 
* tutorial notebook added

* lg tutorial cleaned up

* dataset scaffold

* move dataloader to data

* fix model

* remove todo

* utils seperated

* [model]line graph new implementation + tutorial + binary sub graph dataset

* [tutorial] line graph sphinx scaffold

* [tutorial] lgnn tutorial improved

* [tutorial] remove notebook

* [tutorial] fix lg and gcn links

* [tutorial] fix random seed

* [tutorial]fix

* WIP

* code refactor done

* new mini dataset; remove utils code

* fix

* word fix

* fix link

* minor fix

* minor fix

* minor fix
 2018-12-03 17:59:50 -05:00
Da Zheng 2389df8155 [API] Subgraph sampling API. (#138) 
* add neighbor sampler.

* fix sampler.

* handle non-existing nodes differently in MapParentIdToSubgraphId.

* fix sampling tests.

* parallelize sampling.

* add more tests.

* fix a bug in subgraph loader.

* get number of sampled nodes directly.

* load mxnet CSR to DGLGraph directly.

* create graph from mxnet csr array directly.

* use subgraph loader in SSE.

* fix a bug in sampling.

* output framework tensor.

* fix.

* remove unnecessary code.

* fix tests.

* add SSE README.

* move to contrib.

* address comments.

* add tests.
 2018-11-14 13:51:27 -08:00
GaiYu0 cc372e3774 sbm mixture 2018-10-04 21:22:41 +08:00
Gan Quan b7eb16595f Fix 0deg update_to and Tree-LSTM model (#51) 
* WIP

* WIP

* treelstm dataloader

* Main training loop.

* trainable treelstm script

* fix dependency

* cuda training

* Add tensorized topological traversal

* allowing update_to() with no incoming messages

* fixing partial cases
 2018-08-28 11:57:08 -04:00
Minjie Wang 0a78dbe12a Three DGL-GCN implementations (#35) 
* synthetic dataset

* some fix

* gcn readme
 2018-08-07 15:26:58 -04:00
Minjie Wang 2c489fadec SPMV specialization (#32) 
* fix edge list order problem in cached graph.

* minor fix

* fix bug in edge iter

* SPMV works

* gcn spmv on CPU

* change gcn style

* fix cached graph performance; fixed gcn dataset bug

* reorg dir

* non-batch spmv; partial update problem with shape change

* fix reorder problem; finish gcn-batch impl

* pop API

* GPU context
 2018-08-06 14:00:06 -04:00