【源码】基于深度学习的代码注释自动生成系统

项目简介

本项目作为毕业任务开发，是一个基于深度学习技术的代码注释自动生成系统。其核心目标是输入代码，输出对应的注释，项目涵盖数据处理、模型搭建、预测和评估等流程，采用了seq2seq、transformer等模型。

项目的主要特性和功能

1. 多模型支持：包含seq2seq、seq2seq_onlycode、transformer模型，还实现了seq2trans集成模型。
2. 代码转换：可将代码转换为tensor格式。
3. 预测与评估：能使用训练好的模型进行预测，并生成预测结果和BLEU评分。

安装使用步骤

数据集获取

访问AI Studio上建立的数据集：https://aistudio.baidu.com/aistudio/datasetdetail/73043 。

检查点获取

不同模型的checkpoints获取路径如下： 1. seq2seq模型：https://www.kaggle.com/chenfanliang/seq2seq 2. seq2seq_onlycode模型：https://www.kaggle.com/chenfanliang/seq2seq-onlywith-code 3. transformer模型：https://www.kaggle.com/chenfanliang/transfor

运行项目

1. 确保已下载本项目的源码文件。
2. 运行code2tensor.py将代码转换为tensor。
3. 运行main.py使用模型进行预测。
4. 运行eval.py生成预测结果和BLEU评分。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】