corpora.wikicorpus – Corpus from a Wikipedia dump

corpora.wikicorpus – Corpus from a Wikipedia dump

Construct a corpus from a Wikipedia (or other MediaWiki-based) database dump.

If you have the pattern package installed, this module will use a fancy lemmatization to get a lemma of each token (instead of plain alphabetic tokenizer). The package is available at https://github.com/clips/pattern .

See scripts/process_wiki.py for a canned (example) script based on this module.

class gensim.corpora.wikicorpus.WikiCorpus(fname, processes=None, lemmatize=False, dictionary=None, filter_namespaces=('0', ))

Bases: gensim.corpora.textcorpus.TextCorpus

Treat a wikipedia articles dump (*articles.xml.bz2) as a (read-only) corpus.

The documents are extracted on-the-fly, so that the whole (massive) dump can stay compressed on disk.

>>> wiki = WikiCorpus('enwiki-20100622-pages-articles.xml.bz2') # create word->word_id mapping, takes almost 8h
>>> MmCorpus.serialize('wiki_en_vocab200k.mm', wiki) # another 8h, creates a file in MatrixMarket format plus file with id->word

Initialize the corpus. Unless a dictionary is provided, this scans the corpus once, to determine its vocabulary.

If pattern package is installed, use fancier shallow parsing to get token lemmas. Otherwise, use simple regexp tokenization. You can override this automatic logic by forcing the lemmatize parameter explicitly.

get_texts()

Iterate over the dump, returning text version of each article as a list of tokens.

Only articles of sufficient length are returned (short articles & redirects etc are ignored).

Note that this iterates over the texts; if you want vectors, just use the standard corpus interface instead of this function:

>>> for vec in wiki_corpus:
>>>     print(vec)

getstream()

load(fname, mmap=None)

Load a previously saved object from file (also see save).

If the object was saved with large arrays stored separately, you can load these arrays via mmap (shared memory) using mmap=’r’. Default: don’t use mmap, load large arrays as normal objects.

If the file being loaded is compressed (either ‘.gz’ or ‘.bz2’), then mmap=None must be set. Load will raise an IOError if this condition is encountered.

save(*args, **kwargs)

save_corpus(fname, corpus, id2word=None, metadata=False)

Save an existing corpus to disk.

Some formats also support saving the dictionary (feature_id->word mapping), which can in this case be provided by the optional id2word parameter.

>>> MmCorpus.save_corpus('file.mm', corpus)

Some corpora also support an index of where each document begins, so that the documents on disk can be accessed in O(1) time (see the corpora.IndexedCorpus base class). In this case, save_corpus is automatically called internally by serialize, which does save_corpus plus saves the index at the same time, so you want to store the corpus with:

>>> MmCorpus.serialize('file.mm', corpus) # stores index as well, allowing random access to individual documents

Calling serialize() is preferred to calling save_corpus().

gensim.corpora.wikicorpus.extract_pages(f, filter_namespaces=False)

Extract pages from a MediaWiki database dump = open file-like object f.

Return an iterable over (str, str, str) which generates (title, content, pageid) triplets.

gensim.corpora.wikicorpus.filter_wiki(raw)

Filter out wiki mark-up from raw, leaving only text. raw is either unicode or utf-8 encoded string.

gensim.corpora.wikicorpus.get_namespace(tag)

Returns the namespace of tag.

gensim.corpora.wikicorpus.process_article(args)

Parse a wikipedia article, returning its content as a list of tokens (utf8-encoded strings).

gensim.corpora.wikicorpus.remove_file(s)

Remove the ‘File:’ and ‘Image:’ markup, keeping the file caption.

Return a copy of s with all the ‘File:’ and ‘Image:’ markup replaced by their corresponding captions. See http://www.mediawiki.org/wiki/Help:Images for the markup details.

gensim.corpora.wikicorpus.remove_markup(text)

gensim.corpora.wikicorpus.remove_template(s)

Remove template wikimedia markup.

Return a copy of s with all the wikimedia markup template removed. See http://meta.wikimedia.org/wiki/Help:Template for wikimedia templates details.

Note: Since template can be nested, it is difficult remove them using regular expresssions.

gensim.corpora.wikicorpus.tokenize(content)

Tokenize a piece of text from wikipedia. The input string content is assumed to be mark-up free (see filter_wiki()).

Return list of tokens as utf8 bytestrings. Ignore words shorted than 2 or longer that 15 characters (not bytes!).