From 0945c699c786cba4c75ad5127bf317d15d260349 Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Tue, 1 Jan 2019 15:07:02 +0100 Subject: [PATCH] Fixes #25 by adding article title in markdown format to wiki text --- ulmfit/create_wikitext.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/create_wikitext.py b/ulmfit/create_wikitext.py index 942475a..30073d4 100644 --- a/ulmfit/create_wikitext.py +++ b/ulmfit/create_wikitext.py @@ -24,7 +24,7 @@ def get_texts(root): if text.strip() == title: # print('No content continuing...') continue - yield text + yield (f"={title}=\n"+text) def write_wikitext(file_path, text_iter, mt, num_tokens, mode='w'):