arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2104.01002cs.SEcs.LG
收藏

HAConvGNN: Hierarchical Attention Based Convolutional Graph Neural Network for Code Documentation Generation in Jupyter Notebooks

University of Waterloo(滑铁卢大学);IBM Research(IBM研究院);University of Michigan(密歇根大学);IBM Research Europe(IBM欧洲研究院);JD.COM Silicon Valley Research Center(京东硅谷研究中心)

Xuye Liu, Dakuo Wang, April Wang, Yufang Hou, Lingfei Wu

英文摘要:

Jupyter notebook allows data scientists to write machine learning code together with its documentation in cells. In this paper, we propose a new task of code documentation generation (CDG) for computational notebooks. In contrast to the previous CDG tasks which focus on generating documentation for single code snippets, in a computational notebook, one documentation in a markdown cell often corresponds to multiple code cells, and these code cells have an inherent structure. We proposed a new model (HAConvGNN) that uses a hierarchical attention mechanism to consider the relevant code cells and the relevant code tokens information when generating the documentation. Tested on a new corpus constructed from well-documented Kaggle notebooks, we show that our model outperforms other baseline models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑