graphorge.gnn_base_model.train.training.read_loss_history_from_file

read_loss_history_from_file(loss_record_path)[source]

Read training process loss history from loss history record file.

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

Detaches loss values from computation graph and moves them to CPU.

Parameters:

loss_record_path (str) – Loss history record file path.

Returns:

  • loss_nature (str) – Loss nature.

  • loss_type (str) – Loss function type.

  • training_loss_history (list[float]) – Training process training loss history (per epoch).

  • validation_loss_history ({None, list[float]}) – Training process validation loss history. Set to None if not available.