| --- |
| library_name: newsrex |
| tags: |
| - news-recommendation |
| - unknown |
| - jax |
| - mind |
| datasets: |
| - mind |
| license: apache-2.0 |
| --- |
| |
| # NewsReX unknown — JAX — MIND-small |
|
|
| unknown news recommendation model trained on MIND-small using the |
| [NewsReX](https://github.com/igor17400/NewsReX) framework (JAX). |
|
|
| ## Test Results |
|
|
| | Seed | AUC | MRR | NDCG@5 | NDCG@10 | |
| |------|-----|-----|--------|---------| |
| | 123 * | 0.6747 | 0.3214 | 0.3556 | 0.4193 | |
| | 42 | 0.6716 | 0.3195 | 0.3534 | 0.4178 | |
| | 456 | 0.6738 | 0.3198 | 0.3547 | 0.4184 | |
| | **mean ± std** | **0.6734±0.0013** | **0.3202±0.0009** | **0.3546±0.0009** | **0.4185±0.0006** | |
|
|
| \* Best seed (weights at repo root) |
|
|
| ## Repository Structure |
|
|
| ``` |
| newsrex/unknown-JAX-MIND-small/ |
| ├── model.safetensors ← best seed (123) |
| ├── test_results.json |
| ├── training_run_summary.json |
| ├── seed_123/model.safetensors |
| ├── seed_42/model.safetensors |
| ├── seed_456/model.safetensors |
| └── README.md |
| ``` |
|
|
| ## Usage |
|
|
| ```bash |
| git clone https://github.com/igor17400/NewsReX.git |
| cd NewsReX && uv sync |
| |
| # Run evaluation with best seed weights |
| uv run python src/eval.py \ |
| experiment=mind/unknown \ |
| framework=jax \ |
| weights=hf://newsrex/unknown-JAX-MIND-small/model.safetensors |
| |
| # Run evaluation with a specific seed |
| uv run python src/eval.py \ |
| experiment=mind/unknown \ |
| framework=jax \ |
| weights=hf://newsrex/unknown-JAX-MIND-small/seed_42/model.safetensors |
| ``` |
|
|
| ## Citation |
|
|
| ```bibtex |
| @misc{newsrex2026, |
| title={NewsReX: An Open-Source Multi-Framework for Neural News Recommendation}, |
| author={Igor L. R. Azevedo and Toyotaro Suzumura and Yuichiro Yasui}, |
| year={2025}, |
| eprint={2508.21572}, |
| archivePrefix={arXiv}, |
| primaryClass={cs.IR}, |
| url={https://arxiv.org/abs/2508.21572}, |
| } |
| ``` |
|
|
