Skip to content
Snippets Groups Projects
Commit cffe54c3 authored by Martyna Wiącek's avatar Martyna Wiącek
Browse files

fixed passing relation_distribution and relation_label_distribution

parent a08dc86d
Branches
Tags
1 merge request!46Merge COMBO 3.0 into master
...@@ -109,7 +109,8 @@ class COMBO(PredictorModule): ...@@ -109,7 +109,8 @@ class COMBO(PredictorModule):
relation_distribution, relation_label_distribution) = self._predictions_as_tree(prediction, instance) relation_distribution, relation_label_distribution) = self._predictions_as_tree(prediction, instance)
sentence = conllu2sentence( sentence = conllu2sentence(
tree, sentence_embedding, embeddings, tree, sentence_embedding, embeddings,
relation_distribution, relation_label_distribution relation_distribution=relation_distribution,
relation_label_distribution=relation_label_distribution
) )
sentences.append(sentence) sentences.append(sentence)
return sentences return sentences
...@@ -125,7 +126,11 @@ class COMBO(PredictorModule): ...@@ -125,7 +126,11 @@ class COMBO(PredictorModule):
predictions = super().predict_instance(instance) predictions = super().predict_instance(instance)
(tree, sentence_embedding, embeddings, (tree, sentence_embedding, embeddings,
relation_distribution, relation_label_distribution) = self._predictions_as_tree(predictions, instance) relation_distribution, relation_label_distribution) = self._predictions_as_tree(predictions, instance)
return conllu2sentence(tree, sentence_embedding, embeddings, relation_distribution, relation_label_distribution) return conllu2sentence(
tree, sentence_embedding, embeddings,
relation_distribution=relation_distribution,
relation_label_distribution=relation_label_distribution
)
@overrides @overrides
def predict_json(self, inputs: JsonDict) -> data.Sentence: def predict_json(self, inputs: JsonDict) -> data.Sentence:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment