Skip to content
This repository has been archived by the owner on Jan 29, 2020. It is now read-only.
/ code2vec Public archive

A PyTorch implementation of `code2vec: Learning Distributed Representations of Code` (Alon et al., 2018)

License

Notifications You must be signed in to change notification settings

bentrevett/code2vec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

  1. ./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.
  1. python run.py

To-Do

  • Graph of results
  • Inference code
  • Embedding exploration

About

A PyTorch implementation of `code2vec: Learning Distributed Representations of Code` (Alon et al., 2018)

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published