This repository has been archived by the owner on Jan 29, 2020. It is now read-only.

PyTorch code2vec

This repo contains an implementation of code2vec: Learning Distributed Representations of Code.

Requirements

Python 3+
PyTorch 1.0
A CUDA compatible GPU

Quickstart

./download_preprocessed.sh to get the pre-processed datasets from the code2vec and the code2seq papers.

Note: by default the 3 datasets from code2seq are commented out.

python run.py

To-Do

Graph of results
Inference code
Embedding exploration