Skip to content

rel-trial: study outcome (Classification)

This example now materializes the official rel-trial tables from the relbench Python library and runs the GraphReduce pipeline on those local parquet files.

from relbench_trial_study_outcome import run_rel_trial_study_outcome

df_val, df_test, in_time_auc, holdout_auc, n_features, materialized, target = run_rel_trial_study_outcome()
print(materialized)
print(target, in_time_auc, holdout_auc, n_features, len(df_val), len(df_test))

Current implementation:

Interactive Runner