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. self.metadata if set to true will ensure that serialize will write out article titles to a pickle file.

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()

Yield documents from the underlying plain text collection (of one or more files). Each item yielded from this method will be considered a document by subsequent preprocessing methods.

init_dictionary(dictionary)

If dictionary is None, initialize to an empty Dictionary, and then if there is an input for the corpus, add all documents from that input. If the dictionary is already initialized, simply set it as the corpus’s dictionary.

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.

preprocess_text(text)

Apply preprocessing to a single text document. This should perform tokenization in addition to any other desired preprocessing steps.

Parameters:text (str) – document text read from plain-text file. Returns:tokens produced from text as a result of preprocessing. Return type:iterable of str

sample_texts(n, seed=None, length=None)

Yield n random documents from the corpus without replacement.

Given the number of remaining documents in a corpus, we need to choose n elements. The probability for the current element to be chosen is n/remaining. If we choose it, we just decrease the n and move to the next element. Computing the corpus length may be a costly operation so you can use the optional parameter length instead.

Parameters:

• n (int) – number of documents we want to sample.
• seed (int|None) – if specified, use it as a seed for local random generator.
• length (int|None) – if specified, use it as a guess of corpus length. It must be positive and not greater than actual corpus length.

Yields:

list[str] – document represented as a list of tokens. See get_texts method.

Raises:

ValueError – when n is invalid or length was set incorrectly.

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().

step_through_preprocess(text)

Yield tuples of functions and their output for each stage of preprocessing. This is useful for debugging issues with the corpus preprocessing pipeline.

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.init_to_ignore_interrupt()

Should only be used when master is prepared to handle termination of child processes.

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!).