Bert

Used to generate Bert features for queries and documents

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)