bugfinder.features.extraction.node2vec.embeddings

class bugfinder.features.extraction.node2vec.embeddings.Node2VecEmbeddings(dataset, deprecation_warning=None)

Bases: AbstractProcessing

Class responsible for generating the embeddings, using the Word2Vec model trained with the output of the node2vec algorithm.

embedding_length = 50

execute(**kwargs)

Run the processing. This function reads again the dataset looking for the edges files generated by Joern, process them and generates a embedding vector for each instance.

Returns

dictionary containing all embeddings

Return type

embeddings(dict)

vector_length = 64