supercoderhawk
diff --git a/Collapse file
‎python/dnlp/core/dnn_crf.py‎
Copy file name to clipboardExpand all lines: python/dnlp/core/dnn_crf.py
+1-1Lines changed: 1 addition & 1 deletion b/Collapse file
‎python/dnlp/core/dnn_crf.py‎
Copy file name to clipboardExpand all lines: python/dnlp/core/dnn_crf.py
+1-1Lines changed: 1 addition & 1 deletion
@@ -45,7 +45,7 @@ def __init__(self, *, config: DnnCrfConfig = None, data_path: str = '', dtype: t
     self.output = self.get_output_layer(self.hidden_layer)
 
     if mode == 'predict':
-      self.output = tf.squeeze(self.output, axis=2)
+      self.output = tf.squeeze(tf.transpose(self.output), axis=2)
     elif train == 'll':
       self.ll_loss, _ = tf.contrib.crf.crf_log_likelihood(self.output, self.real_indices, self.seq_length,
                                                           self.transition)