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.

Sign up for GitHub

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

Jump to bottom

add 2 new stackex tasks; 4 amazon tasks #81

Merged
merged 7 commits into from
Jan 30, 2024
Merged

add 2 new stackex tasks; 4 amazon tasks #81

merged 7 commits into from
Jan 30, 2024

Conversation

joshr17
Copy link
Collaborator

@joshr17 joshr17 commented Jan 28, 2024 •
edited
Loading

This PR adds:

Stackex:

  • a new node level task for stackex. The task is to predict whether a user will be awarded a badge within the next 2 years.
  • stackex-votes changed from regression target to binary classification target: simply predict if a post will receive any votes at all. The regression version was highly unbalanced and hard to train models on.
  • existing engage task unchanged

Amazon:

  • customer churn
  • product churn
  • customer LTV (measured by count - simplifying choice for now)
  • product LTV (measured by count - simplifying choice for now)

Baseline performances checked and look good.

@joshr17 joshr17 changed the title add stackex-badges task and make stackex-votes binary add 3 stackex tasks; 4 amazon tasks Jan 30, 2024
@joshr17 joshr17 changed the title add 3 stackex tasks; 4 amazon tasks add 2 new stackex tasks; 4 amazon tasks Jan 30, 2024
@joshr17 joshr17 requested review from weihua916 and removed request for kexinhuang12345 January 30, 2024 00:06
weihua916
weihua916 approved these changes Jan 30, 2024
View reviewed changes
examples/gnn.py Outdated Show resolved Hide resolved
examples/xgboost_baseline.py Outdated Show resolved Hide resolved
examples/xgboost_baseline.py Outdated Show resolved Hide resolved
examples/xgboost_baseline.py Show resolved Hide resolved
@joshr17 joshr17 merged commit 579fc1e into main Jan 30, 2024
4 checks passed
@joshr17 joshr17 deleted the node_tasks branch January 30, 2024 16:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@weihua916 weihua916 weihua916 approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@joshr17 @weihua916