igor174's picture
Upload MINER (jax) trained on MIND-small β€” 3 seeds
fdd9710 verified
---
library_name: newsrex
tags:
- news-recommendation
- miner
- jax
- mind
datasets:
- mind
license: apache-2.0
---
# NewsReX MINER β€” JAX β€” MIND-small
MINER 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.6572 | 0.3102 | 0.3423 | 0.4066 |
| 42 | 0.6554 | 0.3094 | 0.3417 | 0.4059 |
| 456 * | 0.6610 | 0.3154 | 0.3493 | 0.4114 |
| **mean Β± std** | **0.6579Β±0.0024** | **0.3117Β±0.0027** | **0.3444Β±0.0035** | **0.4080Β±0.0025** |
\* Best seed (weights at repo root)
## Repository Structure
```
newsrex/MINER-JAX-MIND-small/
β”œβ”€β”€ model.safetensors ← best seed (456)
β”œβ”€β”€ 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/miner \
framework=jax \
weights=hf://newsrex/MINER-JAX-MIND-small/model.safetensors
# Run evaluation with a specific seed
uv run python src/eval.py \
experiment=mind/miner \
framework=jax \
weights=hf://newsrex/MINER-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},
}
```