spec2vec.utils module

class spec2vec.utils.ModelSaver(num_of_epochs: int, iterations: List, filename: str)

Bases: gensim.models.callbacks.CallbackAny2Vec

Callback to save model during training (when specified).

__init__(num_of_epochs: int, iterations: List, filename: str)

Parameters

• num_of_epochs – Total number of training epochs.

• iterations – Number of total iterations or list of iterations at which to save the model.

• filename – Filename to save model.

on_epoch_end(model)

Allow saving model during training when specified in iterations.

class spec2vec.utils.TrainingProgressLogger(num_of_epochs: int)

Bases: gensim.models.callbacks.CallbackAny2Vec

Callback to log training progress.

__init__(num_of_epochs: int)

Parameters

num_of_epochs – Total number of training epochs.

on_epoch_end(model)

Return progress of model training