labml-python-autocomplete 0.0.3

A simple model that learns to predict Python source code

Source Code Modeling

This repo trains deep learning models on source code.

Try it yourself

1. Clone this repo
2. Install requirements from requirements.txt
3. Run python_autocomplete/create_dataset.py.
   • It collects repos mentioned in PyTorch awesome list
   • Downloads the zip files of the repos
   • Extract the zips
   • Remove non python files
   • Collect all python code to data/train.py and, data/eval.py
4. Run python_autocomplete/train.py to train the model. Try changing hyper-parameters like model dimensions and number of layers.
5. Run evaluate.py to evaluate the model.