Word2Vec

Used to generate Word2Vec features for queries and documents, also supports weighting by tfidf scores

class src.embeddings.bert.Bert

Bases: object

A class to create bert embeddings.

Methods: transform(raw_texts: pd.Series, store: str = None):

Transforms series of unpreprocessed strings to bert embeddings

transform(raw_texts: pandas.core.series.Series, store: Optional[str] = None)

Transform Series of unpreprocessed strings to bert embeddings.

Parameters

raw_texts (pd.Series) – Series of unpreprocessed strings

Returns

List containing bert embeddings

Return type

bert_vec (list)