We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8b6f439 commit ea17a0cCopy full SHA for ea17a0c
rasa/utils/tensorflow/crf.py
@@ -154,9 +154,7 @@ def crf_decode(
154
# argmax tag and the max activation.
155
def _single_seq_fn():
156
decode_tags = tf.cast(tf.argmax(potentials, axis=2), dtype=tf.int32)
157
- decode_scores = tf.reshape(
158
- tf.reduce_max(tf.nn.softmax(potentials, axis=2), axis=2), shape=[-1]
159
- )
+ decode_scores = tf.reduce_max(tf.nn.softmax(potentials, axis=2), axis=2)
160
best_score = tf.reshape(tf.reduce_max(potentials, axis=2), shape=[-1])
161
return decode_tags, decode_scores, best_score
162
0 commit comments