木易 發自 凹非寺
量子位 報導 | 公眾號 QbitAI
又一款懶人神器問世了：
Visual Studio Code的擴展，基於CodeBERT的Python文檔字符串生成器。
看來現在，這群偷「懶」的程式設計師們連文檔字符串都不想自己寫了。
基於CodeBERT的生成器
跟正常的DocStrings用法一樣，你只需要輸入三引號「「」」」，之後按Enter，便能調用這個工具。就像這樣：
並且，這個生成器還可以在幾種不同類型的文檔字符串格式之間進行選擇。
而這個工具，不僅僅能夠幫助寫代碼的人，還可以方便後來人讀代碼。可以說是非常方便的一個擴展了。
這個工具的本質，就是一個自然語言識別的AI，它能夠識別你所寫代碼的內容，然後輸出對應的字符串。
具體是如何識別的，該AI的作者是這麼介紹的：
參數類型是通過PEP 484類型、默認值和var標籤進行識別的。
識別輸入之後便是推斷，這之間進行訓練的方法，也並不複雜，通過CodeBERT便可以實現。
CodeBERT是一種雙模預訓練模型，它可以捕捉自然語言和程式語言之間的語義連接，是目前已知的第一個大型 NL-PL（自然語言-程式語言）預訓練模型。
在訓練過程中，使用 CodeSearchNet 語料庫作為訓練數據，並使用 CodeBERT 中的 Code2NL微調任務。
之後，進行的推斷基準測試如下：
從上圖也能看出，在不同的推斷基準測試中，所需要的生成時間都很短（2s內），這體現在操作上的話，就是延遲。
目前，這個生成器可以支持識別args、kwargs、decorators和errors等參數類型。
此外，還可以輸出Google、docBlockr、Numpy、Sphinx等Docstring格式，PEP0257類型的格式，也即將上線。
使用方法
1、從容器中運行模型推斷服務
具有GPU的話：在安裝nvidia-docker後，運行docker run-it-d-GPU 0-p 5000:5000 gray kode/ai-docstring。
只有CPU：運行docker run-it-d-p 5000:5000 gray kode/ai-docstring。
2、在VSCode中安裝擴展並使用
光標必須在定義正下方的行上，以生成完全自動填充的文檔字符串。
用三重引號「「」」或『』』」打開文檔字符串後，按Enter鍵；
鍵盤快捷鍵是ctrl+shift+2，Mac為cmd+shift+2。
網友討論
該生成器一經發布，便引起了網友們的討論。
有網友認為這是一項非常便利的技術，非常期待：
不過，也有網友認為這項技術仍具有不小的局限性：
比如這位網友認為：更常見的注釋，是在代碼中沒有立即出現的上下文之間添加內容。
不過，雖然有局限，但是人總是在「懶」的過程中變得更「懶」（狗頭）。
比如jQuery的流行，是因為開發者懶得為DOM編寫跨瀏覽器兼容性代碼；
而之後Angular.js的流行，是因為開發者連DOM都懶得操作。
所以，還是非常期待這個擴展的後續，這些程式設計師能不能針對這些局限性進一步偷懶的。
獲取資源
目前這個擴展已經在Visual Studio Code上免費上線。
進入連結即可獲取資源：
https://marketplace.visualstudio.com/items?itemName=graykode.ai-docstring
也可以通過Github獲取源文件：
https://github.com/graykode/ai-docstring
— 完 —
本文系網易新聞•網易號特色內容激勵計劃籤約帳號【量子位】原創內容，未經帳號授權，禁止隨意轉載。
原標題：《Python文檔字符串生成器：基於CodeBERT，支持Google、Numpy等多種輸出格式》
閱讀原文