immuneML.encodings.word2vec.model_creator package¶

Submodules¶

immuneML.encodings.word2vec.model_creator.KmerPairModelCreator module¶

class immuneML.encodings.word2vec.model_creator.KmerPairModelCreator.KmerPairModelCreator[source]¶

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)[source]¶

immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy module¶

class immuneML.encodings.word2vec.model_creator.ModelCreatorStrategy.ModelCreatorStrategy[source]¶

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)[source]¶

immuneML.encodings.word2vec.model_creator.ModelType module¶

class immuneML.encodings.word2vec.model_creator.ModelType.ModelType(value)[source]¶

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[source]¶

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)[source]¶

Module contents¶