def __init__(self, data_signature: Dict[(Text, Dict[(Text, List[FeatureSignature])])], label_data: RasaModelData, entity_tag_specs: Optional[List[EntityTagSpec]], config: Dict[(Text, Any)]) -> None:
    super().__init__('DIET', config, data_signature, label_data)
    self._entity_tag_specs = self._ordered_tag_specs(entity_tag_specs)
    self.predict_data_signature = {feature_name: features for (feature_name, features) in data_signature.items() if (TEXT in feature_name)}
    self._create_metrics()
    self._update_metrics_to_log()
    self.all_labels_embed: Optional[tf.Tensor] = None
    self._prepare_layers()
def __init__(self, data_signature: Dict[(Text, Dict[(Text, List[FeatureSignature])])], label_data: RasaModelData, entity_tag_specs: Optional[List[EntityTagSpec]], config: Dict[(Text, Any)]) -> None:
    super().__init__('DIET', config, data_signature, label_data)
    self._entity_tag_specs = self._ordered_tag_specs(entity_tag_specs)
    self.predict_data_signature = {feature_name: features for (feature_name, features) in data_signature.items() if (TEXT in feature_name)}
    self._create_metrics()
    self._update_metrics_to_log()
    self.all_labels_embed: Optional[tf.Tensor] = None
    self._prepare_layers()