Skip to content

Commit 128bbc7

Browse files
committed
feat: update notebooks, upload HF artifacts, compile paper with citations
- Add real-model loading notes to quickstart and custom_env notebooks - All notebooks have Colab badges pointing to correct URLs - Upload step_0100000.pt checkpoint + config.json to HuggingFace - Full paper recompilation with all citations resolved - Only trivial 5pt URL overflow remains in conclusion
1 parent 27390f5 commit 128bbc7

58 files changed

Lines changed: 3418 additions & 1701 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

CLAUDE.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ The planned package lives at `src/retail_world_model/`. See `docs/project-bluepr
8989

9090
**Stochastic latent**: 32 independent categorical distributions × 32 classes = 1024-dim one-hot. Entity-factored variant: 8 categoricals × 8 classes per entity. Always apply 1% unimix: `probs = 0.99 × softmax(logits) + 0.01/32`.
9191

92-
**Causal constrained decoder**: Per-category price elasticities (`θ̂`) pre-estimated via DML-PLIV are **frozen** in `CausalDemandDecoder`. The MLP residual learns everything else. This prevents the world model from learning a confounded price-demand relationship. Expected elasticity range: 2.0 to −3.0 (grocery). See `docs/project-blueprint.md §9`.
92+
**Causal constrained decoder**: Per-category price elasticities (`θ̂`) pre-estimated via DML-PLIV are **frozen** in `CausalDemandDecoder`. The MLP residual learns everything else. This prevents the world model from learning a confounded price-demand relationship. Expected elasticity range: ~−0.9 for shelf-stable categories (canned soup); −2.0 to −3.0 for more elastic categories (beer, soft drinks). See `docs/project-blueprint.md §9`.
9393

9494
**KL balancing**: `β_pred=1.0`, `β_dyn=0.5` (sg(posterior)), `β_rep=0.1` (sg(prior)), free bits=1 nat. The 5:1 asymmetry prevents posterior collapse.
9595

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,10 @@ dreamprice/
170170
└── docs/ # Data, results, documentation
171171
```
172172

173+
## Future Work
174+
175+
- Inference-time price optimization via CEM or gradient-based planning (`inference/planning.py`)
176+
173177
## License
174178

175179
CC-BY-NC-4.0. Data from Kilts Center for Marketing, University of Chicago.

0 commit comments

Comments
 (0)