镜像站点 · 本页由第三方 GitHub 只读镜像提供,非 GitHub 官方站点,不接受任何登录或凭据输入。前往 github.com
Skip to content

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

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

simon824 merged 33 commits into
apache:mainfrom
edw4rdyao:integrate-dgl

Conversation

@edw4rdyao

@edw4rdyao edw4rdyao commented Sep 20, 2024 •

Copy link
Copy Markdown
Contributor

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
Comment thread hugegraph-ml/setup.py Resolved
Comment thread hugegraph-ml/requirements.txt Resolved
Comment thread hugegraph-ml/README.md Resolved
@imbajin imbajin added this to the 1.5.0 milestone Sep 20, 2024

@imbajin imbajin left a comment

Copy link
Copy Markdown
Member

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

Comment thread hugegraph-ml/README.md Resolved
Comment thread hugegraph-ml/README.md Resolved
@edw4rdyao

Copy link
Copy Markdown
Contributor Author

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

imbajin commented Sep 23, 2024 •

Copy link
Copy Markdown
Member

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 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
@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
haohao0103 pushed a commit to haohao0103/incubator-hugegraph-ai that referenced this pull request Jun 30, 2026
…s based on DGL (apache#83)

* feat(ml): add the basic structure of integrated DGL

* feat(ml): add jknet&grand examples and uni-test

* doc(ml): add requirements.txt

* doc(ml): update README.md

* chore: fix review suggestions

* fix(ml): add license header for pyproject.toml

* fix(ml): unix-style cmd path & reformat the files

* feat(ml): enable convert multi-graph hugegraph2dgl & dgl2hugegraph

* fix(ml): set default user to "" & add pylint ci for hugegraph-ml

* fix(ml): dgl&torch version in requirements.txt for linux

* fix(ml): fix dependency & pylint

* feat(ml): encapsulate convert a DGL graph by vertices and edges

* feat(ml): enable convert a graph dataset from hugegraph

* refact(ml): do not return graph info dict

The relevant information of a single graph is obtained directly through DGLGraph

* refact(ml): change param {DATASET}_graph_vertex for convert graph

* test(ml): add examples test

add example test
specifies that the dataset is uppercase
update README.md

* fix: "str | None" for py3.9

* feat(ml): add diffpool, gin(global_pool) model and example

* fix(ml): test teardown & pylint

* fix(ml): torch version vulnerability

* feat(ml): add import hetero graph to hg (utils)

* feat(ml): enable convert hetero graph from hugegraph

* refact(ml): move train/val/test mask data to vertex props

* feat(ml): batch import graph data from dgl

---------

Co-authored-by: imbajin <jin@apache.org>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

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

Projects

Status: ✅ Done

Development

Successfully merging this pull request may close these issues.

3 participants