Exemplo n.º 1
0
 def create_policy(self, featurizer, priority):
     # use standard featurizer from EmbeddingPolicy,
     # since it is using FullDialogueTrackerFeaturizer
     p = EmbeddingPolicy(priority=priority,
                         attn_before_rnn=True,
                         attn_after_rnn=True)
     return p
Exemplo n.º 2
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(
         featurizer=featurizer,
         priority=priority,
         **{"scale_loss": False, "evaluate_on_num_examples": 4},
     )
     return p
Exemplo n.º 3
0
 def create_policy(self, featurizer, priority):
     # use standard featurizer from EmbeddingPolicy,
     # since it is using FullDialogueTrackerFeaturizer
     p = EmbeddingPolicy(priority=priority, **tf_defaults())
     return p
Exemplo n.º 4
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer,
                         priority=priority,
                         **tf_defaults())
     return p
Exemplo n.º 5
0
 def create_policy(self, featurizer, priority):
     # use standard featurizer from EmbeddingPolicy,
     # since it is using MaxHistoryTrackerFeaturizer
     # if max_history is specified
     p = EmbeddingPolicy(priority=priority, max_history=self.max_history)
     return p
Exemplo n.º 6
0
 def create_policy(self, featurizer, priority):
     # use standard featurizer from EmbeddingPolicy,
     # since it is using FullDialogueTrackerFeaturizer
     # if max_history is not specified
     p = EmbeddingPolicy(priority=priority)
     return p
Exemplo n.º 7
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer,
                         priority=priority,
                         **{"loss_type": "margin"})
     return p
Exemplo n.º 8
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer, priority=priority)
     return p
Exemplo n.º 9
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(
         featurizer=featurizer, priority=priority, **{"ranking_length": 11}
     )
     return p