基于思维链的软件漏洞自动修复技术研究

环境需求:

请安装以下程序以满足CotRepair的基本运行需求:

    pytorch=2.0.0;
    datasets==1.16.1;
    transformers==4.21.1;
    nltk=3.8.1;
    scipy=1.10.1;

结果复现:

使用如下命令可以训练CotRepair:

bash scripts/CotRepair.sh -g [GPU_ID]

训练完毕后,使用如下命令可以评估CotRepair:

bash scripts/CotRepair.sh -g [GPU_ID] -e

CotRepair的运行结果以及基线方法的结果已存于`output`文件夹下.