kkcookies99 / UAST

implement for UAST network
MIT License
8 stars 1 forks source link

UAST

implement for UAST network

requirements:

pytorch

treesitter

description:

data_loader_XXXX: dataloader of those networks

SAST: Sequence-based AST network

GAST: Graph-based AST network

UAST: Unified AST network

gen_graph: generate graph from AST

pre_ast_path: get AST from source code

unified_vocab: generate unified vocabulary