TextData.calculate_default_embeddings#

TextData.calculate_default_embeddings(model: str = 'miniLM', file_path: str = 'embeddings.csv')[source]#

Calculate the default properties of the dataset.

Parameters
modelstr, default: ‘miniLM’

The model to use for calculating the embeddings. Possible values are: ‘miniLM’: using the miniLM model in the sentence-transformers library. ‘open_ai’: using the ADA model in the open_ai library. Requires an API key.

file_pathstr, default: ‘embeddings.csv’

The path to save the embeddings to.

previous

TextData.calculate_builtin_properties

next

TextData.cast_to_dataset