Amazon EV/CM Implementation

This repository implements an EV/CM (Moe–Fader style) model for Amazon browsing and purchase sessions.

Run

python -m src.evcm_pipeline \
  --input data/amazon_sessions.csv \
  --output reports/evcm \
  --visit-unit daily \
  --freq W \
  --n-sims 300 \
  --ev-starts 5 \
  --cm-starts 5 \
  --seed 123

Tests

pytest

JAX/CUDA12 run

# Optional: install GPU JAX wheel
JAX_ACCELERATOR=cuda12 bash scripts/setup_codex_jax.sh

python -m src.evcm_pipeline \
  --input data/amazon_sessions.csv \
  --output reports/evcm_jax \
  --visit-unit daily \
  --freq W \
  --n-sims 300 \
  --ev-starts 5 \
  --cm-starts 5 \
  --engine jax \
  --x64 \
  --seed 123