load_embeddings#

load_embeddings(as_train_test: bool = True) Union[array, Tuple[array, array]][source]#

Load and return the embeddings of the tweet_emotion dataset calculated by OpenAI.

Parameters
as_train_testbool, default: True

If True, the returned data is split into train and test exactly like the toy model was trained. The first return value is the train data and the second is the test data. Otherwise, returns a single object.

Returns
embeddingsnp.ndarray

Embeddings for the tweet_emotion dataset.