phages2050.embeddings.nucleotides package

Submodules

phages2050.embeddings.nucleotides.word2vec module

class phages2050.embeddings.nucleotides.word2vec.Word2VecEmbedding(model_pkl_file: str)[source]

Bases: object

Word2Vec instance loader class

get_train_params() → Exception[source]

TODO: return dict with model train parameters

class phages2050.embeddings.nucleotides.word2vec.Word2VecModelManager(model_dir: str = 'word2vec_model')[source]

Bases: object

Manager class is responsible to download and unzip Word2Vec pre-trained model for nucleotides embedding

STATUS_CODE_200 = 200
WORD2VEC_URL = b'https://deeppetri.ai/static/phages2050/word2vec-embedding-21.07.2020.zip'
download_model() → pathlib.Path[source]

Download Word2Vec pre-trained model and unzip it into directory

This procedure should be executed once and the result loaded by Word2VecEmbedding class instance

Module contents