From 2fc4448a476b734b4d775cef30b9c64195a6b662 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martyna=20Wi=C4=85cek?= <martyna.wiacek@ipipan.waw.pl>
Date: Mon, 20 Nov 2023 22:12:50 +0100
Subject: [PATCH] changed softmax to inner product in matrix
---
combo/models/parser.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/combo/models/parser.py b/combo/models/parser.py
index bb0fc91..9b38be8 100644
--- a/combo/models/parser.py
+++ b/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"])
--
GitLab