rasa.core.featurizers.single_state_featurizer
SingleStateFeaturizer Objects
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__
Initialize the single state featurizer.
get_entity_tag_ids
Returns the tag to index mapping for entities.
Returns:
Tag to index mapping.
prepare_for_training
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 the given state with the help of the given interpreter.
Arguments:
state
- The state to encodeinterpreter
- The interpreter used to encode the state
Returns:
A dictionary of state_type to list of features.
encode_entities
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 locationsinterpreter
- The interpreter used to encode the state
Returns:
A dictionary of entity type to list of features.
encode_all_actions
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.