def create_policy(self, featurizer):
// use standard featurizer from EmbeddingPolicy,
// since it is using FullDialogueTrackerFeaturizer
p = EmbeddingPolicy()
return p
@pytest.fixture(scope="module")
def trained_policy(self, featurizer):
After Change
def create_policy(self, featurizer):
// use standard featurizer from EmbeddingPolicy,
// since it is using FullDialogueTrackerFeaturizer
p = EmbeddingPolicy(attn_before_rnn=True, attn_after_rnn=False)
return p
class TestEmbeddingPolicyAttentionAfterRNN(PolicyTestCollection):