torch.imdb

Contents

torch.imdb#

Module: `torch.imdb`#

Objects helpful in analysis of IMDB data from keras. Constructs a lookup table that is usable directly with keras.datasets.imdb data.

Functions#

ISLP.torch.imdb.load_lookup(root='.')#

ISLP.torch.imdb.load_sequential(root='.')#: Load IMDB features from ISLP package in torch sparse tensor format.

ISLP.torch.imdb.load_sparse(root='.', validation=2000, random_state=0)#: Load IMDB features from ISLP package in scipy.sparse.coo_matrix format.

ISLP.torch.imdb.load_tensor(root='.')#: Load IMDB features from ISLP package in torch sparse tensor format.