Fix span score logging

This commit is contained in:
Paul O'Leary McCann 2022-04-13 20:37:06 +09:00
parent d470fa03c1
commit 2300f4df3d

View File

@ -415,7 +415,7 @@ DEFAULT_SPAN_PREDICTOR_MODEL = Config().from_str(default_span_predictor_config)[
"input_prefix": "coref_head_clusters",
"output_prefix": "coref_clusters",
},
default_score_weights={"span_predictor_f": 1.0, "span_predictor_p": None, "span_predictor_r": None},
default_score_weights={"span_accuracy": 1.0},
)
def make_span_predictor(
nlp: Language,