Skip to content

Latest commit

 

History

History
16 lines (9 loc) · 680 Bytes

README.md

File metadata and controls

16 lines (9 loc) · 680 Bytes

Tensorflow Implementation of Tree-based Convolutional Neural Network

Implementation of TBCNN using Tensorflow. Our report here.

Instead of training on C programs, This version is used for SQL injection detection. Dataset is provided in the data folder.

Installation

It's highly recommended to install into a virtual environment. Run the following command after activating venv:

pip install -r requirements/dev.txt

Usage

tbcnn.embedding is the entry point for pretraining the embedding matrix.

tbcnn.tbcnn is the entry point for the tbcnn model. It requires a pre-trained embedding matrix.