Skip to content
Snippets Groups Projects
Commit b78828d6 authored by kreuzer's avatar kreuzer
Browse files

Aktualisieren models.py

parent 3a0e5678
No related branches found
No related tags found
No related merge requests found
......@@ -75,7 +75,7 @@ class SummarisationModel(nn.Module):
r_1, r_2, r_l = utils.rouge(utils.select_elements(datapoint.raw_document, top_indices), datapoint.raw_summary, verbose=True)
running_rouge_1 += r_1
running_rouge_2 += r_2
running_rouge_L += r_l
running_rouge_l += r_l
epoch_rouge_1 = running_rouge_1 / len(dataset)
epoch_rouge_2 = running_rouge_2 / len(dataset)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment