赛道一 A 榜方案复现

当前记录的 A 榜最高分为 1.3919313477799322。

对应方法由 Dataset1 的候选来源结构重排和 Dataset2 的二部图传播预混合重排组成。

环境

审核环境为 Ubuntu 22.04、Python 3.10、CUDA 12.4 和 Jittor 1.3.10。安装依赖时执行：

python3 -m pip install -r requirements.txt

程序使用系统中的 CUDA 和 C++ 编译器。一般不需要设置额外变量；只有工具不在系统搜索路径中时，才需要显式设置 nvcc_path 或 cc_path。

数据

将原始数据放在以下位置：

data/dataset1/train.csv
data/dataset1/test.csv
data/dataset2/train.csv
data/dataset2/test.csv

程序会在训练开始前检查四个文件及表头，不读取任何测试标签。

完整复现

从原始数据训练全部模型并生成提交包：

python3 main.py --rebuild-all --data-root data --output-dir outputs/rebuild_all

输出文件为：

outputs/rebuild_all/result_current_best.zip

该压缩包只包含 dataset1.csv 和 dataset2.csv。实测 Dataset2 的全部中间资产约为 9.1 GB，其中增强测试缓存约为 3.8 GB；建议至少预留 15 GB 可用磁盘空间。完整复现要求输出目录为空，避免误用旧训练产物。

训练完成后，可使用本次训练写出的配置重新执行推理：

python3 main.py --config outputs/rebuild_all/rebuild_config.toml

所有训练代码均使用 Jittor。仓库和审核包中不包含 PyTorch 依赖，也不包含历史测试预测文件。