Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add GLEM model, TAGDataset and example of GLEM #9662

Open
wants to merge 20 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 8 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
696fe7a
add GLEM model, TAGDataset and example of GLEM
ECMGit Sep 15, 2024
430c4fd
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Sep 15, 2024
78d9781
fix docstring unexpected intentation
ECMGit Sep 15, 2024
a22742c
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Sep 15, 2024
c74cbcb
Merge branch 'master' into gnn-llm-integration-glem
puririshi98 Sep 24, 2024
599e3d8
Merge branch 'master' into gnn-llm-integration-glem
puririshi98 Oct 8, 2024
558bac9
Merge branch 'master' into gnn-llm-integration-glem
puririshi98 Oct 9, 2024
10a4e40
Merge branch 'master' into gnn-llm-integration-glem
akihironitta Oct 16, 2024
96fe9b5
Merge branch 'master' into gnn-llm-integration-glem
puririshi98 Oct 23, 2024
a13cafa
change data folder and remove test acc metrics in each epoch
ECMGit Oct 26, 2024
471b602
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 26, 2024
09301da
remove unused variables
ECMGit Oct 27, 2024
8ef78f0
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 27, 2024
9377881
precommit fix
ECMGit Oct 27, 2024
745ed21
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 27, 2024
320bb80
Merge branch 'master' into gnn-llm-integration-glem
puririshi98 Oct 29, 2024
7c22ec0
select best model by validation accuracy
ECMGit Oct 29, 2024
c5db5cf
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 29, 2024
f56cbce
precommit failure fix
ECMGit Oct 29, 2024
62784d3
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Oct 29, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).

### Added

- Added `nn.models.GLEM` ([#9662](https://github.com/pyg-team/pytorch_geometric/pull/9662))
- Added `TAGDataset` ([#9662](https://github.com/pyg-team/pytorch_geometric/pull/9662))
- Consolidated `examples/ogbn_{papers_100m,products_gat,products_sage}.py` into `examples/ogbn_train.py` ([#9467](https://github.com/pyg-team/pytorch_geometric/pull/9467))

### Changed
Expand Down
11 changes: 8 additions & 3 deletions examples/llm/README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
# Examples for Co-training LLMs and GNNs

| Example | Description |
| ------------------------------------ | ----------------------------------------------------------------------------------------------------------------------------------------------------------- |
| [`g_retriever.py`](./g_retriever.py) | Example for Retrieval-Augmented Generation (RAG) w/ GNN+LLM by co-training `LLAMA2` with `GAT` for answering questions based on knowledge graph information |
| Example | Description |
| ------------------------------------ | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
| [`g_retriever.py`](./g_retriever.py) | Example for Retrieval-Augmented Generation (RAG) w/ GNN+LLM by co-training `LLAMA2` with `GAT` for answering questions based on knowledge graph information |
| [`glem.py`](./glem.py) | Example for [GLEM](https://arxiv.org/abs/2210.14709), a GNN+LLM co-training model via variational Expectation-Maximization (EM) framework on node classification tasks to achieve SOTA results |

## Run GLEM for getting SOTA result on ogbn-products dataset

`python glem.py`
ECMGit marked this conversation as resolved.
Show resolved Hide resolved
Loading
Loading