sgnlp.models.ufd.utils.create_dataset_embedding

create_dataset_embedding(cfg: sgnlp.models.ufd.data_class.UFDArguments, dataset_type: str) → Dict

Main helper wrapper function to generate datasets.

Parameters

• cfg (UFDArguments) – UFDArguments config load from configuration file

• dataset_type (str) – type of dataset to generate embeddings

Raises

Exception – raise exception for unknown dataset types

Returns

return dictionary of dataset embeddings.

Return type

Dict