fast.ai ULMFiT callback to extract sentence embeddings
 

class SentenceEmbeddingCallback[source]

SentenceEmbeddingCallback(pool_mode='concat') :: Callback

Basic class handling tweaks of the training loop by changing a Learner in various events

Test Sentence Embedding Callback

Fine-Tune model on AG_NEWS dataset, extract embeddings and visualize them with matplotlib