機器之心報道

編輯：魔王

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

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

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

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

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

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

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

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

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

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

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

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

• 如果只有 CPU：利用 Google Colab 和 ngrok 運行 flask 服務(wù)器(推薦)或者使用 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。他對開源文化感興趣，認為一個好的項目可以衍生新的行業(yè)和技術(shù)。個人主頁：https://libraries.io/github/graykode

Amazon SageMaker?1000元大禮包

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

??THE END?

轉(zhuǎn)載請聯(lián)系本公眾號獲得授權(quán)

投稿或?qū)で髨蟮?#xff1a;content@jiqizhixin.com

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。