newsrex/TCCM-JAX-MIND-small

NewsReX TCCM — JAX — MIND-small

TCCM news recommendation model trained on MIND-small using the NewsReX framework (JAX).

Test Results

Seed	AUC	MRR	NDCG@5	NDCG@10
123	0.6781	0.3256	0.3639	0.4251
42	0.6809	0.3251	0.3635	0.4253
456 *	0.6834	0.3321	0.3694	0.4308
mean ± std	0.6808±0.0022	0.3276±0.0032	0.3656±0.0027	0.4271±0.0026

* Best seed (weights at repo root)

Repository Structure

newsrex/TCCM-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

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/tccm \
    framework=jax \
    weights=hf://newsrex/TCCM-JAX-MIND-small/model.safetensors

# Run evaluation with a specific seed
uv run python src/eval.py \
    experiment=mind/tccm \
    framework=jax \
    weights=hf://newsrex/TCCM-JAX-MIND-small/seed_42/model.safetensors

Citation

@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},
}