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]