Skip to content

CAAI BDSC2023 CUP 任务1《社交图谱小样本场景链接预测》,初复赛第一名的代码,主要思路基于知识图谱

License

Notifications You must be signed in to change notification settings

ralgond/Codes-for-CAAI-BDSC2023-CUP-1st-place-solution

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

项目介绍

https://tianchi.aliyun.com/competition/entrance/532073

汇报PPT

https://github.com/ralgond/Codes-for-CAAI-BDSC2023-CUP-1st-place-solution/blob/main/BDSC2023.pptx

解题思路是基于知识图谱中的链接预测技术。

软硬件环境

PyTorch 1.11.0

Python 3.8(ubuntu20.04)

Cuda 11.3

RTX 3090(24GB) * 1

部署与运行

部署

unzip AREA.zip

cd AREA

pip install -r requirements.txt

mkdir -p ./src/raw_data/ecom-social/

mkdir -p ./src/data/ecom-social/

mkdir -p ./src/save/ecom-social/

将初赛和复赛的文件都放到 ./src/raw_data/ecom-social/

这些文件如下:

event_info.json
source_event_preliminary_train_info.json
target_event_final_test_info.json
target_event_final_train_info.json
target_event_preliminary_test_info.json
target_event_preliminary_train_info.json
user_info.json

运行代码:

cd src

python process_data2.py

bash run.sh
# 程序大概会运行 2.5 个小时...

cd ..

程序运行结束

结果文件会保存在 ./submit/, 文件名为submit.json

关于结果的随机性

结果会在一定范围里波动,请一直执行run.sh,直到复现最优成绩即可。

About

CAAI BDSC2023 CUP 任务1《社交图谱小样本场景链接预测》,初复赛第一名的代码,主要思路基于知识图谱

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published