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)

Bases: immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy.ModelCreatorStrategy

create_model(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, k: int, vector_size: int, batch_size: int, model_path: pathlib.Path, sequence_type: immuneML.environment.SequenceType.SequenceType)

immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy module

class immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy.ModelCreatorStrategy(epochs: int, window: int)

Bases: object

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

abstract create_model(dataset: immuneML.data_model.dataset.RepertoireDataset.RepertoireDataset, k: int, vector_size: int, batch_size: int, model_path: pathlib.Path, sequence_type: immuneML.environment.SequenceType.SequenceType)

immuneML.encodings.word2vec.model_creator.ModelType module

class immuneML.encodings.word2vec.model_creator.ModelType.ModelType(value)

Bases: enum.Enum

An enumeration.

KMER_PAIR = 2

SEQUENCE = 1

immuneML.encodings.word2vec.model_creator.SequenceModelCreator module

class immuneML.encodings.word2vec.model_creator.SequenceModelCreator.SequenceModelCreator(epochs: int, window: int)

Bases: immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy.ModelCreatorStrategy

create_model(dataset: immuneML.data_model.dataset.Dataset.Dataset, k: int, vector_size: int, batch_size: int, model_path: pathlib.Path, sequence_type: immuneML.environment.SequenceType.SequenceType)

Module contents