Add summarization name mapping for MultiNews (#18117)

* Add summarization name mapping for MultiNews * Add summarization name mapping for MultiNews

Add summarization name mapping for MultiNews (#18117)
* Add summarization name mapping for MultiNews * Add summarization name mapping for MultiNews
fde22c75 · John Giorgi · GitHub · 19513336 · fde22c75 · fde22c75
Unverified Commit fde22c75 authored Jul 13, 2022 by John Giorgi Committed by GitHub Jul 13, 2022
2 changed files
--- a/examples/pytorch/summarization/run_summarization.py
+++ b/examples/pytorch/summarization/run_summarization.py
@@ -286,6 +286,7 @@ summarization_name_mapping = {
    "xglue": ("news_body", "news_title"),
    "xsum": ("document", "summary"),
    "wiki_summary": ("article", "highlights"),
+    "multi_news": ("document", "summary"),
 }

--- a/examples/tensorflow/summarization/run_summarization.py
+++ b/examples/tensorflow/summarization/run_summarization.py
@@ -266,6 +266,7 @@ summarization_name_mapping = {
    "xglue": ("news_body", "news_title"),
    "xsum": ("document", "summary"),
    "wiki_summary": ("article", "highlights"),
+    "multi_news": ("document", "summary"),
 }
 # endregion