Beispiel #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
Beispiel #2
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(
         featurizer=featurizer,
         priority=priority,
         **{"scale_loss": False, "evaluate_on_num_examples": 4},
     )
     return p
Beispiel #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
Beispiel #4
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer,
                         priority=priority,
                         **tf_defaults())
     return p
Beispiel #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
Beispiel #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
Beispiel #7
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer,
                         priority=priority,
                         **{"loss_type": "margin"})
     return p
Beispiel #8
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(featurizer=featurizer, priority=priority)
     return p
Beispiel #9
0
 def create_policy(self, featurizer, priority):
     p = EmbeddingPolicy(
         featurizer=featurizer, priority=priority, **{"ranking_length": 11}
     )
     return p