Example of a recommender system #8546

pmpalang · 2023-12-05T22:57:41Z

Hi Everyone,

I'm adding a recommender system example with the following salient features

Dataset MovieLens – a heterogenous use case
Demonstrates the use of edge based temporal sampling
Visualization
t-SNE based visualization (--visualize_emb)
Uses torch_geometric.nn.pool.MIPSKNNIndex for getting recommedations
Integration of the LinkPred metrics -- precision@k and ndcg@k

Thanks,
Poovaiah

for more information, see https://pre-commit.ci

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

…into pyg-team-master

…ranch

for more information, see https://pre-commit.ci

examples/hetero/recommender_system.py

…orch_geometric into recommender_example

for more information, see https://pre-commit.ci

…orch_geometric into recommender_example

for more information, see https://pre-commit.ci

pmpalang and others added 30 commits October 18, 2023 10:32

enabling timestamp in movielens dataset

bd85574

Merge branch 'master' of https://github.com/pyg-team/pytorch_geometric

f4bc07f

Merge branch 'master' of https://github.com/pyg-team/pytorch_geometric

6991f82

Enabling edge-based temporal sampling

f65c72f

Merge branch 'master' of https://github.com/pmpalang/pytorch_geometric

eafd0e4

Cleaning up code, adding profile_code flag to examples

72c7f30

[pre-commit.ci] auto fixes from pre-commit.com hooks

73b907d

for more information, see https://pre-commit.ci

Merge branch 'master' of https://github.com/pmpalang/pytorch_geometric

3ce9bae

[pre-commit.ci] auto fixes from pre-commit.com hooks

58d5351

for more information, see https://pre-commit.ci

Update torch_geometric/datasets/movie_lens.py

a892d2f

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

Update torch_geometric/sampler/neighbor_sampler.py

23b300a

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

Update examples/homo_link_pred_temporal.py

c7cc84b

Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>

Updated changelog.md

c7bf62e

Merge branch 'master' of https://github.com/pmpalang/pytorch_geometric

72b7aec

Merge branch 'master' of https://github.com/pyg-team/pytorch_geometric …

948665f

…into pyg-team-master

Merge branch 'master' of https://github.com/pyg-team/pytorch_geometric …

6c6a57b

…into pyg-team-master

Merge branch 'master' of https://github.com/pyg-team/pytorch_geometric …

a204833

…into pyg-team-master

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

fa9d548

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

38fc341

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

b3f188e

Adding a recommender example using MovieLens dataset

5ee2dcf

removing e2e examples for temporal -- they should be in a different b…

7cb81b1

…ranch

spaces

f5a9c8e

spaces

ca5af98

update

61a68ed

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

8a7f7e6

update

3f627c3

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

781a105

Merge branch 'pyg-team-master' into recommender_example

26544c2

recommendation example based on edge temporal e2e example

4617d40

pmpalang and others added 2 commits December 8, 2023 13:04

Added the exclude edges feature in MIPSKNNIndex to the recsys example

517be60

[pre-commit.ci] auto fixes from pre-commit.com hooks

19f8915

for more information, see https://pre-commit.ci

rusty1s reviewed Dec 11, 2023

View reviewed changes

examples/hetero/recommender_system.py Outdated Show resolved Hide resolved

pmpalang and others added 16 commits December 11, 2023 12:33

Making inference and evaluation mini batch based

df998be

Merge branch 'recommender_example' of https://github.com/pmpalang/pyt…

1666334

…orch_geometric into recommender_example

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

1fab815

Merge branch 'pyg-team-master' into recommender_example

9c20dd1

[pre-commit.ci] auto fixes from pre-commit.com hooks

f944a43

for more information, see https://pre-commit.ci

PEP8

d030d59

Merge branch 'recommender_example' of https://github.com/pmpalang/pyt…

ef56bfd

…orch_geometric into recommender_example

update

3381dd7

temporal split strategy

52e4f85

time should be 1 less than the edge_time

d8893cd

edge_time -1

896ba39

Merge branch 'recommender_example' of https://github.com/pmpalang/pyt…

cc4584d

…orch_geometric into recommender_example

Changelog

0cd0c50

Merge https://github.com/pyg-team/pytorch_geometric into pyg-team-master

aba9d47

Merge branch 'pyg-team-master' into recommender_example

6dbc249

[pre-commit.ci] auto fixes from pre-commit.com hooks

cbbc96a

for more information, see https://pre-commit.ci

rusty1s added the feature label Jan 1, 2024

rusty1s added 3 commits February 11, 2024 11:11

update

d76410d

Merge branch 'master' into recommender_example

b6de8dd

update

4f154c4

rusty1s approved these changes Feb 11, 2024

View reviewed changes

rusty1s added 3 commits February 11, 2024 11:17

update

8a18ce9

update

bac4c67

update

33d83f4

rusty1s enabled auto-merge (squash) February 11, 2024 11:25

update

eccc11f

rusty1s merged commit 123e38e into pyg-team:master Feb 11, 2024
13 checks passed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Example of a recommender system #8546

Example of a recommender system #8546

pmpalang commented Dec 5, 2023 •

edited

Example of a recommender system #8546

Example of a recommender system #8546

Conversation

pmpalang commented Dec 5, 2023 • edited

pmpalang commented Dec 5, 2023 •

edited