Tensorflow 代码自动补全工具

标签:   PythonJavaFavorite

Tensorflow 代码自动补全工具是基于循环神经网络(RNN)的能对Tensorflow代码进行自动补全的小工具。

这个项目主要包含两个方面:服务器端和客户端。在服务器端中,我们实现了一个RNN,他可以从token层次对tensorflow代码进行学习。数据集是从Github上进行爬取,经过筛选和预处理后输入到神经网络之中进行训练。训练完成后RNN就可以对一段给定的tensorflow代码来预测接下来的token,其准确率高达71%。至于客户端,我们主要开发了一个Eclipse插件,它自定义了一个Python的编辑器,能过对用户的输入等进行处理,将数据发送到服务器并从服务器接收到返回,显示在屏幕上给予用户提示。

服务器端使用Tensorflow编写,网络有两层,每层有512的LSTM的节点。客户端则使用Java编写,并可以作为插件嵌入到Eclipse之中。

客户端的项目是开源的,你也可以观看Demo视频来了解这个工具是如何来帮助用户完成Tensorflow代码的。