Closed egoriyaa closed 2 months ago
Add possibility to load pretrained embedding models
For TS2VecEmbeddingModel and TSTCCEmbeddingModel classes modify load method in such way:
TS2VecEmbeddingModel
TSTCCEmbeddingModel
load
def load(cls, path: Optional[pathlib.Path] = None, model_name: Optional[str] = None)
path
model_name
etna
For TS2VecEmbeddingModel and TSTCCEmbeddingModel classes add static method list_models
list_models
For TS2VecEmbeddingModel and TSTCCEmbeddingModel add parameter freezed in constructor. - If freezed is True, freeze model in constructor.
freezed
freeze
Modify 210-embedding_models notebook to show how to use pretrained models
210-embedding_models
For load method:
list_models returns right model names
No response
🚀 Task description
Add possibility to load pretrained embedding models
Plan
For
TS2VecEmbeddingModel
andTSTCCEmbeddingModel
classes modifyload
method in such way:path
path
is specified,model_name
is not - download the local model frompath
path
is not specified,model_name
is specified - download the pretrained model to theetna
folder in the home directoryFor
TS2VecEmbeddingModel
andTSTCCEmbeddingModel
classes add static methodlist_models
For
TS2VecEmbeddingModel
andTSTCCEmbeddingModel
add parameterfreezed
in constructor. - Iffreezed
is True,freeze
model in constructor.Modify
210-embedding_models
notebook to show how to use pretrained modelsTest cases
For
load
method:model_name
not inlist_models
list_models
returns right model namesAdditional context
No response