Create virtual environment: ./venv.sh
Activate virtual environment: source env/bin/activate
Proposed models are working with AST so there is a possibility to complete any language. For now there is possibility to test model on two datasets:
- Javascript (js150 dataset link)
- Python (py150 dataset link)
To train model on Javascript dataset:
- Download data:
./scripts/ast/data_download.sh
- Process data:
./scripts/ast/data_process.sh
- Train model:
./scripts/ast/run.sh
To change model parameters edit file: scripts/ast/train.sh
To train model on Python dataset:
- Download data:
./scripts/pyast/data_download.sh
- Process data:
./scripts/pyast/data_process.sh
- Train model:
./scripts/pyast/run.sh
To change model parameters edit file: scripts/pyast/train.sh
For accuracy visualization tensorboard is used. To run it use: ./scripts/tensorboard.sh