Struct Code2VecEncoder 

pub struct Code2VecEncoder { /* private fields */ }

Code2Vec encoder for generating embeddings from AST paths

Implementations

impl Code2VecEncoder

pub fn new(dim: usize) -> Code2VecEncoder

Create a new Code2Vec encoder

pub fn with_seed(self, seed: u64) -> Code2VecEncoder

Set random seed for reproducible embeddings

pub fn dim(&self) -> usize

Get the embedding dimension

pub fn encode_path(&self, path: &AstPath) -> Vec<f64>

Encode a single path to an embedding vector

pub fn aggregate_paths(&self, paths: &[AstPath]) -> CodeEmbedding

Aggregate multiple path embeddings into a single code embedding

pub fn add_token_embedding( &mut self, token: impl Into<String>, embedding: Vec<f64>, )

Add a token embedding to the vocabulary

pub fn add_path_embedding( &mut self, path_key: impl Into<String>, embedding: Vec<f64>, )

Add a path embedding to the vocabulary

pub fn token_vocab_size(&self) -> usize

Get vocabulary size for tokens

pub fn path_vocab_size(&self) -> usize

Get vocabulary size for paths

Trait Implementations

impl Debug for Code2VecEncoder

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for Code2VecEncoder

fn default() -> Code2VecEncoder

Returns the “default value” for a type.