Views
No views yet
| Seed | AUC | MRR | NDCG@5 | NDCG@10 |
|---|---|---|---|---|
| 42 * | 0.6681 | 0.3302 | 0.3692 | 0.4272 |
1model:
2 name: pprec
3 architecture:
4 news_encoder:
5 type: multi_head_self_attention
6 news_dim: 400
7 num_heads: 20
8 head_dim: 20
9 co_num_heads: 5
10 co_head_dim: 40
11 attention_hidden_dim: 200
12 entity_embedding_dim: 100
13 category_embedding_dim: 200
14 use_entity: true
15 use_recency: true
16 use_ctr: true
17 use_activity_gate: true
18 popularity_embedding_bins: 200
19 popularity_embedding_dim: 400
20 recency_embedding_bins: 1500
21 recency_embedding_dim: 100
22 ctr_scaler_init: 19.0
23 pop_content_dims:
24 - 256
25 - 256
26 - 128
27 pop_recency_dims:
28 - 64
29 - 64
30 pop_gate_dims:
31 - 128
32 - 64
33 activity_gate_hidden_dim: 64
34 embedding:
35 size: 300
36 trainable: true
37 dropout_rate: 0.2
38 seed: 42
39inputs:
40 title:
41 max_length: 32
42 history:
43 max_length: 50
44 impressions:
45 max_length: 5
46 max_entities: 5
47 process_title: true
48 process_abstract: false
49 process_category: true
50 process_subcategory: false
51 process_user_id: false
52training:
53 loss:
54 name: categorical_crossentropy
55 from_logits: true
56 reduction: sum_over_batch_size
57 label_smoothing: 0.0
58 optimizer: adam
59 learning_rate: 0.0001
60 batch_size: 256
61 num_epochs: 10
62 early_stopping:
63 patience: 3
64 min_improvement: 0.01
65 negative_sampling:
66 strategy: random
67 candidates: 4
68evaluation:
69 mode: fast
70 evaluator: pp_rec
71 metrics:
72 - auc
73 - mrr
74 - ndcg@5
75 - ndcg@10
76 batch_size: 256newsrex/PPREC-JAX-MIND-small-bert-random/
├── model.safetensors ← best seed (42)
├── test_results.json
├── training_run_summary.json
├── seed_42/model.safetensors
└── README.md1git clone https://github.com/igor17400/NewsReX.git
2cd NewsReX && uv sync
3
4# Run evaluation with best seed weights
5uv run python src/eval.py \
6 experiment=mind/glove/pprec \
7 framework=jax \
8 weights=hf://newsrex/PPREC-JAX-MIND-small-bert-random/model.safetensors
9
10# Run evaluation with a specific seed
11uv run python src/eval.py \
12 experiment=mind/glove/pprec \
13 framework=jax \
14 weights=hf://newsrex/PPREC-JAX-MIND-small-bert-random/seed_42/model.safetensors1@misc{newsrex2026,
2 title={NewsReX: An Open-Source Multi-Framework for Neural News Recommendation},
3 author={Igor L. R. Azevedo and Toyotaro Suzumura and Yuichiro Yasui},
4 year={2025},
5 eprint={2508.21572},
6 archivePrefix={arXiv},
7 primaryClass={cs.IR},
8 url={https://arxiv.org/abs/2508.21572},
9}