:mod:`sourced.ml.core.models.id2vec` ==================================== .. py:module:: sourced.ml.core.models.id2vec Module Contents --------------- .. py:class:: Id2Vec Bases::class:`modelforge.Model` id2vec model - source code identifier embeddings. .. attribute:: NAME :annotation: = id2vec .. attribute:: VENDOR :annotation: = source{d} .. attribute:: DESCRIPTION :annotation: = Model that contains information on source code as identifier embeddings. .. attribute:: LICENSE .. attribute:: embeddings :class:`numpy.ndarray` with the embeddings of shape (N tokens x embedding dims). .. attribute:: tokens List with the processed source code identifiers. .. method:: construct(self, embeddings, tokens) .. method:: dump(self) .. method:: items(self) Returns the tuples belonging to token -> index mapping.