Skip to content

rel-amazon: user churn (Classification)

This example now sources data from the relbench Python library, not the old S3 parquet mirror. The runner materializes the official rel-amazon tables locally through get_dataset("rel-amazon", download=True) before building the GraphReduce frame.

from relbench_amazon_user_churn import run_rel_amazon_user_churn

df, auc, n_features, materialized, target = run_rel_amazon_user_churn()
print(materialized)
print(target, auc, n_features, len(df))

Current implementation:

Current RelBench-aligned dates:

  • validation cut date: 2015-10-01
  • holdout cut date: 2016-01-01
  • lookback start: 2008-01-01

Interactive Runner