shap.datasets.imdb

shap.datasets.imdb(display=False, n_points=None)

Return the classic IMDB sentiment analysis training data in a nice package.

Full data is at: http://ai.stanford.edu/~amaas/data/sentiment/aclImdb_v1.tar.gz Paper to cite when using the data is: http://www.aclweb.org/anthology/P11-1015