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.