Version: 2.2.x

rasa.core.featurizers.single_state_featurizer

SingleStateFeaturizer Objects

class SingleStateFeaturizer()

Base class to transform the dialogue state into an ML format.

Subclasses of SingleStateFeaturizer will decide how a bot will transform the dialogue state into a dictionary mapping an attribute to its features. Possible attributes are: INTENT, TEXT, ACTION_NAME, ACTION_TEXT, ENTITIES, SLOTS and ACTIVE_LOOP. Each attribute will be featurized into a list of rasa.utils.features.Features.

__init__

| __init__() -> None

Initialize the single state featurizer.

get_entity_tag_ids

| get_entity_tag_ids() -> Dict[Text, int]

Returns the tag to index mapping for entities.

Returns:

Tag to index mapping.

prepare_for_training

| prepare_for_training(domain: Domain, interpreter: NaturalLanguageInterpreter) -> None

Gets necessary information for featurization from domain.

Arguments:

  • domain - An instance of :class:rasa.shared.core.domain.Domain.
  • interpreter - The interpreter used to encode the state

encode_state

| encode_state(state: State, interpreter: NaturalLanguageInterpreter) -> Dict[Text, List["Features"]]

Encode the given state with the help of the given interpreter.

Arguments:

  • state - The state to encode
  • interpreter - The interpreter used to encode the state

Returns:

A dictionary of state_type to list of features.

encode_entities

| encode_entities(entity_data: Dict[Text, Any], interpreter: NaturalLanguageInterpreter) -> Dict[Text, List["Features"]]

Encode the given entity data with the help of the given interpreter.

Produce numeric entity tags for tokens.

Arguments:

  • entity_data - The dict containing the text and entity labels and locations
  • interpreter - The interpreter used to encode the state

Returns:

A dictionary of entity type to list of features.

encode_all_actions

| encode_all_actions(domain: Domain, interpreter: NaturalLanguageInterpreter) -> List[Dict[Text, List["Features"]]]

Encode all action from the domain using the given interpreter.

Arguments:

  • domain - The domain that contains the actions.
  • interpreter - The interpreter used to encode the actions.

Returns:

A list of encoded actions.