val.jsonl
收藏资源简介:
数据集包含用户编号、用户历史观看电影列表(按时间顺序排列,越靠后表示越近期观看)、用户实际观看的下一部电影以及推荐系统召回阶段得到的候选电影列表。
The dataset consists of user IDs, the chronologically ordered list of movies the user has previously watched (with later entries indicating more recently viewed content), the next movie the user actually watched, and the candidate movie list obtained during the recall stage of the recommendation system.
上海创智学院2025年PE考试数据集概述
数据集基本信息
- 用途说明:仅供学习和交流
- 数据文件:val.jsonl
- 安装依赖:
pip install openai
数据结构
python { "user_id": 5737, // 用户编号 "item_list": [ // 用户历史观看电影列表,按时间顺序排列 [1836, "Last Days of Disco, The"], // [电影ID, 电影名称] [3565, "Where the Heart Is"], // ... ], "target_item": [1893, "Beyond Silence"], // 用户实际观看的下一部电影 "candidates": [ // 推荐系统召回阶段得到的候选电影列表 [2492, "20 Dates"], [684, "Windows"], [1893, "Beyond Silence"], // 包含用户实际观看的下一部电影 // ... ] }
评估配置
- 超参数:
api_key: str (必须提供)base_url: str (默认https://api.deepseek.com)model_name: str (默认deepseek-chat)num_epochs: int (默认5)temperature: float (默认0.0)ndcg@k: int (默认k=10)is_multi_turn: bool (默认False)
评估方法
-
执行命令: bash bash eval.sh
或 python python eval.py --api_key <your_api_key> --model_name <your_model_name> --num_epochs <your_num_epochs> --temperature <your_temperature> --ndcg@k your_ndcg@k --is_multi_turn <your_is_multi_turn>
评估结果(NDCG@10)
| Epoch | 平均NDCG@10 |
|---|---|
| 1 | 0.60508 |
| 2 | 0.66818 |
| 3 | 0.66201 |
| 4 | 0.66376 |
| 5 | 0.60946 |
| 最终结果 | 0.64170 |




