Skip to content

feat(ml): node emdedding, node classify and graph classify with models based on DGL#83

Merged
simon824 merged 33 commits intoapache:mainfrom
edw4rdyao:integrate-dgl
Nov 7, 2024
Merged

feat(ml): node emdedding, node classify and graph classify with models based on DGL#83
simon824 merged 33 commits intoapache:mainfrom
edw4rdyao:integrate-dgl

Conversation

@edw4rdyao
Copy link
Contributor

@edw4rdyao edw4rdyao commented Sep 20, 2024 •
edited
Loading

This PR is derived from the project of GLCC2024: https://www.gitlink.org.cn/hugegraph/hugegraph/wiki?wiki=graph-ai-1

Features:

  1. Read graphs from HugeGraph and convert them to DGL format
  • Convert single graph dataset
  • Convert mulltigraph dataset
  • Convert heterogeneous graph dataset
  1. Implemented graph learning algorithms (for algorithim test, please visit hugegraph-integrate-model-test)
  • DGI, GRACE, GRAND, JKNet
  • DiffPool, GlobalPool
  • HAN
  1. Encapsulated training and evaluation to simplify the end-to-end graph learning workflow.
  • Node classify task
  • Graph classify task
  • Node embedding task

Additional Notes:

  1. Since the PGL hasn't been updated for a long time, priority is given to integrating the DGL.

@github-actions github-actions bot added the ml label Sep 20, 2024
imbajin
imbajin reviewed Sep 20, 2024
View reviewed changes
@imbajin imbajin added this to the 1.5.0 milestone Sep 20, 2024
imbajin
imbajin reviewed Sep 22, 2024
View reviewed changes
Copy link
Member

@imbajin imbajin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Also note the py-lint ci failed

image

@edw4rdyao
Copy link
Contributor Author

edw4rdyao commented Sep 22, 2024

Also note the py-lint ci failed

image

pylint failed for the hugegraph-llm and hugegraph-python-client, should we fix it? Also, should we add pylint CI for hugegraph-ml in workflows and code_format_and_analysis.sh?

@imbajin
Copy link
Member

imbajin commented Sep 23, 2024 •
edited
Loading

pylint failed for the hugegraph-llm and hugegraph-python-client, should we fix it? Also, should we add pylint CI for hugegraph-ml in workflows and code_format_and_analysis.sh?

It's a bit strange, let me take a look (Most of the issues should indeed be fixed, and if they are unnecessary, they should be eliminated)

edw4rdyao and others added 14 commits September 23, 2024 22:50
@edw4rdyao
feat(ml): enable convert multi-graph hugegraph2dgl & dgl2hugegraph
4352414
@edw4rdyao
fix(ml): set default user to "" & add pylint ci for hugegraph-ml
b73e479
@edw4rdyao
Merge branch 'integrate-dgl' into multigraph
57f6d76
@edw4rdyao
fix(ml): dgl&torch version in requirements.txt for linux
1526727
@edw4rdyao
Merge branch 'integrate-dgl' into multigraph
ccc16f0
@imbajin
Merge branch 'main' into integrate-dgl
ac71c9e
@edw4rdyao
Merge branch 'integrate-dgl' into multigraph
03f9256
@edw4rdyao
fix(ml): fix dependency & pylint
af6f73c
@edw4rdyao
Merge branch 'multigraph' of github.com:edw4rdyao/incubator-hugegraph…
c16ca59 
…-ai into multigraph
@edw4rdyao
feat(ml): encapsulate convert a DGL graph by vertices and edges
50daaf9
@edw4rdyao
feat(ml): enable convert a graph dataset from hugegraph
c2eef47
@edw4rdyao
refact(ml): do not return graph info dict
633b68b 
The relevant information of a single graph is obtained directly through DGLGraph
@edw4rdyao
refact(ml): change param {DATASET}_graph_vertex for convert graph
69ee75b
@edw4rdyao
test(ml): add examples test
32fd88f 
add example test
specifies that the dataset is uppercase
update README.md
@edw4rdyao edw4rdyao changed the title feat(ml): node emdedding&classify with models(DGI, JKNet, GRAND, GRACE) based on DGL feat(ml): node emdedding, node classify and graph classify with models based on DGL Sep 28, 2024
@imbajin imbajin mentioned this pull request Sep 28, 2024
Open
25 tasks
@simon824 simon824 marked this pull request as ready for review November 5, 2024 09:49
@dosubot dosubot bot added size:XXL This PR changes 1000+ lines, ignoring generated files. enhancement New feature or request labels Nov 5, 2024
@simon824 simon824 merged commit 08a649e into apache:main Nov 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@imbajin imbajin imbajin left review comments

@simon824 simon824 simon824 approved these changes

Assignees

No one assigned

Labels

enhancement New feature or request ml size:XXL This PR changes 1000+ lines, ignoring generated files.

Projects

HugeGraph Tasks
Status: ✅ Done

Milestone

1.5.0

Development

Successfully merging this pull request may close these issues.

3 participants

@edw4rdyao @imbajin @simon824