Graph-Augmented Code Summarization in Computational Notebooks

Abstract

Computational notebooks allow data scientists to express their ideas through a combination of code and documentation. However, data scientists often pay attention only to the code, and neglect the creation of the documentation. In this work, we present a human-centered automation system Themisto that can support users to easily create documentation via three approaches: 1) We have developed and reported a GNN-augmented code documentation generation algorithm in a previous paper, which can generate documentation for a given source code; 2) Themisto implements a query-based approach to retrieve online API documentation as the summary for certain types of source code; 3) Themisto also enables a user prompt approach to motivate users to write documentation for some use cases that automation does not work well.