zoo.examples.textclassification package

Submodules

zoo.examples.textclassification.news20 module

zoo.examples.textclassification.news20.download_glove(dest_dir)

zoo.examples.textclassification.news20.download_news20(dest_dir)

zoo.examples.textclassification.news20.get_glove(base_dir='./data/glove.6B', dim=100)

Parse the pre-trained glove6B word2vec and return a dict mapping from word to vector, given the dim of a vector. The word embeddings will be downloaded automatically if not found in the target base_dir.

zoo.examples.textclassification.news20.get_news20(base_dir='./data/20news-18828')

Parse 20 Newsgroup dataset and return a list of (tokens, label). The dataset will be downloaded automatically if not found in the target base_dir.

zoo.examples.textclassification.text_classification module

Module contents