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 RetailHero and MovieLens25 bipartite datasets with causal parameters #9471

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

geopanag
Copy link

I would like to add two new bipartite networks in PyG.
One is based on the RetailHero (https://ods.ai/competitions/x5-retailhero-uplift-modeling/data) (users-buy-products) with causal outcomes and interventions from a marketing campaign, and the other is based on MovieLens25 (https://grouplens.org/datasets/movielens/25m/) (movies-ratedby-users) with observational causal parameters .

I use them to test a GNN for causal outcome prediction in https://hal.science/hal-04601553/document ( ECMLPKDD 2024 ) and I believe they re useful for this line of research.

@geopanag geopanag requested a review from wsad1 as a code owner June 30, 2024 19:54
Copy link

codecov bot commented Jun 30, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 86.87%. Comparing base (dafbd30) to head (a63ab81).
Report is 37 commits behind head on master.

Additional details and impacted files
@@            Coverage Diff             @@
##           master    #9471      +/-   ##
==========================================
- Coverage   88.25%   86.87%   -1.38%     
==========================================
  Files         473      463      -10     
  Lines       30844    30617     -227     
==========================================
- Hits        27221    26599     -622     
- Misses       3623     4018     +395     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@rusty1s rusty1s changed the title Add retailhero and movielens25 bipartite networks with causal parameters Add RetailHero and MovieLens25 bipartite datasets with causal parameters Jul 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants