changed softmax to inner product in matrix

combo/models/parser.py

@@ -155,8 +155,10 @@ class DependencyRelationModel(base.Predictor):
         output = head_output
         output["embedding"] = dep_rel_pred
         #import pdb;pdb.set_trace()
-        output["deprel_label_distribution"] = F.softmax(relation_prediction[:, 1:, 1:], dim=-1)
-        output["deprel_tree_distribution"] = head_pred_soft
+        # output["deprel_label_distribution"] = F.softmax(relation_prediction[:, 1:, 1:], dim=-1)
+        output["deprel_label_distribution"] = relation_prediction[:, 1:, 1:]
+        # output["deprel_tree_distribution"] = head_pred_soft
+        output["deprel_tree_distribution"] = head_pred
 
         if self.training:
             output["prediction"] = (relation_prediction.argmax(-1)[:, 1:], head_output["prediction"])