autointent.modules.base.BaseEmbedding

class autointent.modules.base.BaseEmbedding

Bases: autointent.modules.base.BaseModule, abc.ABC

Base class for embedding modules.

get_train_data(context)

Get train data.

Parameters:

context (autointent.Context) – Context to get train data from

Returns:

Tuple of train utterances and train labels

Return type:

tuple[list[str], autointent.custom_types.ListOfLabels]