機器之心報道

編輯：魔王

該擴展利用可處理編程語言和自然語言的預訓練模型 CodeBERT，實現快速生成 Python 文檔字符串的功能。

Visual Studio Code(簡稱 VS Code)是一個由微軟開發，同時支持 Windows、 Linux 和 macOS 等操作系統的免費代碼編輯器，它支持測試，并內置了 Git 版本控制功能，同時也具有開發環境功能，例如代碼補全、代碼片段和代碼重構等。VS Code 是程序員常用的代碼編輯器之一，是一個可在所有平臺上使用的開源、可擴展和輕量級的編輯器。這些品質使微軟的 VS Code 大受歡迎，并成為 Python 開發的絕佳平臺。此外，VS Code 還擁有大量擴展，可以幫助提高開發效率、提升開發幸福感。今天，我們要介紹的就是一款可以生成 Python 文檔字符串的 VS Code 擴展。

• GitHub 地址：https://github.com/graykode/ai-docstring

• 擴展地址：https://marketplace.visualstudio.com/items?itemName=graykode.ai-docstring&ssr=false

該擴展利用 AI (NLP) 技術，快速生成 Python 函數的文檔字符串(docstring)。原本 Python 函數的描述必須由用戶編寫，而使用這款擴展之后，AI 可以查看代碼并進行總結。具體而言，該擴展使用了微軟等提出的預訓練模型 CodeBERT，該模型可以處理雙模態數據：自然語言和編程語言。在訓練過程中，該擴展使用 GitHub 團隊創建的 CodeSearchNet 語料庫作為訓練數據，并使用 CodeBERT 中的 Code2NL(代碼到自然語言)微調任務。該擴展的推斷基準如下所示：

特性這款 VS Code 擴展具備以下特性：

• AI 快速生成文檔字符串；

• 可以選擇多種不同類型的文檔字符串格式；

• 通過 pep484 類型提示、默認值和變量名，推斷參數類型；

• 支持 args、kwargs、decorators、errors 和 parameter 類型。

其中支持的文檔字符串格式包括：Google(默認)、docBlockr、Numpy、Sphinx，項目主頁顯示這款擴展還即將支持 PEP0257。如何使用這款 VS Code 擴展的使用較為簡單。用戶只需從容器中運行模型推斷服務器，并在 VS Code 中安裝擴展，即可使用。具體步驟如下所示：1. 從容器中運行模型推斷服務器

• 如果已有 GPU 機器：在安裝 nvidia-docker 后，運行 docker run -it -d --gpus 0 -p 5000:5000 graykode/ai-docstring:gpu；

• 如果只有 CPU：利用 Google Colab 和 ngrok 運行 flask 服務器(推薦)或者使用 docker cpu 鏡像：docker run -it -d -p 5000:5000 graykode/ai-docstring:cpu。

2. 在 VS Code 中安裝擴展并使用光標必須在定義下方，以生成完全自動填充的文檔字符串。

• 使用三引號("""或''')打開文檔字符串，然后按回車鍵；

• 鍵盤快捷鍵：ctrl+shift+2，Mac 電腦的快捷鍵是 cmd+shift+2；

• 命令：Generate Docstring；

• 右擊菜單：Generate Docstring。

項目作者該項目的作者是來自韓國慶熙大學的計算機工程本科生 Tae-Hwan Jung。他對開源文化感興趣，認為一個好的項目可以衍生新的行業和技術。個人主頁：https://libraries.io/github/graykode

Amazon SageMaker?1000元大禮包

ML訓練成本降90%，被全球上萬家公司使用，Amazon SageMaker是全托管機器學習平臺，支持絕大多數機器學習框架與算法，并且用 IDE 寫代碼、可視化、Debug一氣呵成。現在，我們準備了1000元的免費額度，開發者可以親自上手體驗，讓開發高質量模型變得更加輕松。

??THE END?

轉載請聯系本公眾號獲得授權

投稿或尋求報道：content@jiqizhixin.com

總結

以上是生活随笔為你收集整理的vscode python 自动补全_利用CodeBERT，这个VS Code扩展可以自动生成Python文档字符串...的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。