Code AutoComplete
code-autocomplete, a code completion plugin for Python.
code-autocomplete实现了Python代码行粒度和块粒度自动补全功能。
Guide
Feature
Demo
http://42.193.145.218/product/short_text_sim/
Install
pip3 install -U code-autocomplete
or
git clone https://github.com/shibing624/code-autocomplete.git
cd code-autocomplete
python3 setup.py install
Usage
Code Completion
基于GPT-2 model
预测整行代码
import sys
sys.path.append('..')
from autocomplete.gpt2 import predict
if __name__ == '__main__':
prompts = [
"""from torch import nn
class LSTM(Module):
def __init__(self, *,
n_tokens: int,
embedding_size: int,
hidden_size: int,
n_layers: int):""",
"""import numpy as np
import torch
import torch.nn as""",
"import java.util.ArrayList;",
]
for prompt in prompts:
res = predict(prompt, model_dir='shibing624/code-autocomplete-gpt2')
print("\n\n======================\n\n")
print("Query:", prompt)
print("\nResult:, res")
print("\n\n======================\n\n")
Contact
- Issue(建议):
- 邮件我:xuming: [email protected]
- 微信我:
加我微信号:xuming624, 备注:个人名称-公司-NLP 进NLP交流群。
Citation
如果你在研究中使用了code-autocomplete,请按如下格式引用:
@misc{code-autocomplete,
author = {Xu Ming},
title = {code-autocomplete: Code AutoComplete with GPT model},
year = {2022},
publisher = {GitHub},
journal = {GitHub repository},
url = {https://github.com/shibing624/code-autocomplete},
}
License
授权协议为 The Apache License 2.0,可免费用做商业用途。请在产品说明中附加code-autocomplete的链接和授权协议。
Contribute
项目代码还很粗糙,如果大家对代码有所改进,欢迎提交回本项目,在提交之前,注意以下两点:
- 在
tests
添加相应的单元测试 - 使用
python setup.py test
来运行所有单元测试,确保所有单测都是通过的
之后即可提交PR。