Team Ai
Datasetpublic

Merlin041/imdb-distilbert-features

IMDb Movie Reviews - DistilBERT Contextual Embedding Cache This dataset contains pre-extracted contextual embedding features of the standard IMDb Movie Reviews dataset (Sentiment Analysis). The features were extracted using a frozen DistilBERT (distilbert-base-uncased) encoder. By caching these high-dimensional embeddings, you can train downstream classifiers (like LSTMs, GRUs, Attention heads, or custom Transformers) in seconds on a local GPU or CPU, bypassing the massive… See the full description on the dataset page: https://huggingface.co/datasets/Merlin041/imdb-distilbert-features.

sourceHugging Faceupdated 3mo agoView on Hugging Face
0likes78downloads
filetest_feat.npy4.58 GBdownload
filetest_mask.npy3.1 MBdownload
filetrain_feat.npy4.58 GBdownload
filetrain_mask.npy3.1 MBdownload

Merlin041/imdb-distilbert-features · main · files are served by the source, never re-hosted here