graphorge.gnn_base_model.train.training.load_loss_history

load_loss_history(model, loss_nature, loss_type, epoch=None)[source]

Load training process training loss history record.

Loss history record file is stored in model_directory under the name loss_history_record.pkl.

Parameters:

  • model (torch.nn.Module) – Model.

  • loss_nature (str) – Loss nature.

  • loss_type (str) – Loss function type.

  • epoch (int, default=None) – Epoch to which loss history is loaded (included), with the first epoch being 0. If None, then loads the full loss history.

Returns:

training_loss_history – Training process training loss history (per epoch).

Return type:

list[float]