immuneML.encodings.word2vec.model_creator package¶

Submodules¶

immuneML.encodings.word2vec.model_creator.KmerPairModelCreator module¶

class immuneML.encodings.word2vec.model_creator.KmerPairModelCreator.KmerPairModelCreator(epochs: int, window: int)[source]¶

Bases: ModelCreatorStrategy

create_model(dataset: RepertoireDataset, k: int, vector_size: int, batch_size: int, model_path: Path, sequence_type: SequenceType)[source]¶

immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy module¶

class immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy.ModelCreatorStrategy(epochs: int, window: int)[source]¶

Bases: object

Defines how word2vec model can be created by defining different contexts for k-mers

abstract create_model(dataset: RepertoireDataset, k: int, vector_size: int, batch_size: int, model_path: Path, sequence_type: SequenceType)[source]¶

immuneML.encodings.word2vec.model_creator.ModelType module¶

class immuneML.encodings.word2vec.model_creator.ModelType.ModelType(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)[source]¶

Bases: Enum

KMER_PAIR = 2¶

SEQUENCE = 1¶

immuneML.encodings.word2vec.model_creator.SequenceModelCreator module¶

class immuneML.encodings.word2vec.model_creator.SequenceModelCreator.SequenceModelCreator(epochs: int, window: int)[source]¶