├── .gitignore ├── .gitpod.yml ├── README.md ├── __init__.py ├── argument_parser.py ├── fast_pb2.py ├── install.sh ├── requirements.txt ├── script ├── download_data.py ├── process_data.py └── process_data.sh ├── tbcnn_testing_script.sh ├── tbcnn_training_script.sh ├── test_tbcnn.py ├── test_torch.py ├── train_tbcnn.py ├── util ├── __init__.py ├── data │ ├── __init__.py │ ├── data_loader │ │ ├── __init__.py │ │ └── base_data_loader.py │ └── data_processor │ │ ├── __init__.py │ │ ├── base_data_processor.py │ │ ├── c_txl_data_processor.py │ │ ├── pycparser_data_processor.py │ │ ├── rust_txl_data_processor.py │ │ ├── srcml_data_processor.py │ │ ├── treesitter_c_data_processor.py │ │ └── treesitter_data_processor.py ├── identifier_splitting.py ├── network │ ├── base_layer.py │ └── tbcnn.py ├── threaded_iterator.py └── util_functions.py └── vocab ├── pycparser ├── node_token │ └── token.txt └── node_type │ └── type.txt ├── srcml ├── node_token │ └── token.txt └── node_type │ └── type.txt └── treesitterc ├── node_token └── token.txt └── node_type └── type.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitpod.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/.gitpod.yml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/__init__.py -------------------------------------------------------------------------------- /argument_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/argument_parser.py -------------------------------------------------------------------------------- /fast_pb2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/fast_pb2.py -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/install.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/script/download_data.py -------------------------------------------------------------------------------- /script/process_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/script/process_data.py -------------------------------------------------------------------------------- /script/process_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/script/process_data.sh -------------------------------------------------------------------------------- /tbcnn_testing_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/tbcnn_testing_script.sh -------------------------------------------------------------------------------- /tbcnn_training_script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/tbcnn_training_script.sh -------------------------------------------------------------------------------- /test_tbcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/test_tbcnn.py -------------------------------------------------------------------------------- /test_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/test_torch.py -------------------------------------------------------------------------------- /train_tbcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/train_tbcnn.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/__init__.py -------------------------------------------------------------------------------- /util/data/data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_loader/__init__.py -------------------------------------------------------------------------------- /util/data/data_loader/base_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_loader/base_data_loader.py -------------------------------------------------------------------------------- /util/data/data_processor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/__init__.py -------------------------------------------------------------------------------- /util/data/data_processor/base_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/base_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/c_txl_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/c_txl_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/pycparser_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/pycparser_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/rust_txl_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/rust_txl_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/srcml_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/srcml_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/treesitter_c_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/treesitter_c_data_processor.py -------------------------------------------------------------------------------- /util/data/data_processor/treesitter_data_processor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/data/data_processor/treesitter_data_processor.py -------------------------------------------------------------------------------- /util/identifier_splitting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/identifier_splitting.py -------------------------------------------------------------------------------- /util/network/base_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/network/base_layer.py -------------------------------------------------------------------------------- /util/network/tbcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/network/tbcnn.py -------------------------------------------------------------------------------- /util/threaded_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/threaded_iterator.py -------------------------------------------------------------------------------- /util/util_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/util/util_functions.py -------------------------------------------------------------------------------- /vocab/pycparser/node_token/token.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/pycparser/node_token/token.txt -------------------------------------------------------------------------------- /vocab/pycparser/node_type/type.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/pycparser/node_type/type.txt -------------------------------------------------------------------------------- /vocab/srcml/node_token/token.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/srcml/node_token/token.txt -------------------------------------------------------------------------------- /vocab/srcml/node_type/type.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/srcml/node_type/type.txt -------------------------------------------------------------------------------- /vocab/treesitterc/node_token/token.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/treesitterc/node_token/token.txt -------------------------------------------------------------------------------- /vocab/treesitterc/node_type/type.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bdqnghi/tbcnn.tensorflow/HEAD/vocab/treesitterc/node_type/type.txt --------------------------------------------------------------------------------