def _check_input_dimension_consistency(self,
model_data: RasaModelData) -> None:
"""Checks if features have same dimensionality if hidden layers are shared."""
if self.component_config.get(SHARE_HIDDEN_LAYERS):
num_text_features = model_data.feature_dimension(TEXT_FEATURES)
num_label_features = model_data.feature_dimension(LABEL_FEATURES)
if num_text_features != num_label_features:
raise ValueError(
"If embeddings are shared text features and label features "
"must coincide. Check the output dimensions of previous components."
)
def test_get_num_of_features(model_data: RasaModelData):
num_features = model_data.feature_dimension("text_features", "sentence")
assert num_features == 24
