SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add summarization name mapping for MultiNews (#18117)

Browse Source 
* Add summarization name mapping for MultiNews

* Add summarization name mapping for MultiNews
This commit is contained in:
John Giorgi
2022-07-13 05:19:20 -07:00
committed by GitHub
parent 195133363e
commit fde22c75a1
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/pytorch/summarization/run_summarization.py
View File

@@ -286,6 +286,7 @@ summarization_name_mapping = {
"xglue": ("news_body", "news_title"), "xglue": ("news_body", "news_title"),
"xsum": ("document", "summary"), "xsum": ("document", "summary"),
"wiki_summary": ("article", "highlights"), "wiki_summary": ("article", "highlights"),
"multi_news": ("document", "summary"),
} }

1
examples/tensorflow/summarization/run_summarization.py
View File

@@ -266,6 +266,7 @@ summarization_name_mapping = {
"xglue": ("news_body", "news_title"), "xglue": ("news_body", "news_title"),
"xsum": ("document", "summary"), "xsum": ("document", "summary"),
"wiki_summary": ("article", "highlights"), "wiki_summary": ("article", "highlights"),
"multi_news": ("document", "summary"),
} }
# endregion # endregion