def from_pre_trained(cls, vocab: Vocab, class_count: int, mode: str,
update_vectors: bool, activation: str):
embedding_bag = EmbeddingBag.from_pretrained(
vocab.vectors, mode=mode, freeze=(not update_vectors))
return cls(embedding_bag, class_count, activation)
def from_pre_trained(cls, vocab: Vocab, class_count: int):
embedding_bag = EmbeddingBag.from_pretrained(vocab.vectors)
return cls(embedding_bag, class_count)