Enhanced dependency parsing develop to master

combo/models/graph_parser.py

@@ -148,6 +148,7 @@ class GraphDependencyRelationModel(base.Predictor):
         output = head_output
 
         output["prediction"] = (relation_prediction.argmax(-1), head_output["prediction"])
+        output["rel_probability"] = relation_prediction
 
         if labels is not None and labels[0] is not None:
             if sample_weights is None: