TrainingHistory

class TrainingHistory

Contains all recorded batch events as a list of BatchTrainingEvent objects and epoch events as a list of EpochTrainingEvent objects.

NOTE: Used to record BatchTrainingEvent and EpochTrainingEvent during training phase.

Constructors

TrainingHistory

fun TrainingHistory()

Functions

appendBatch

fun appendBatch(batchTrainingEvent: BatchTrainingEvent)

Appends one BatchTrainingEvent.

fun appendBatch(epochIndex: Int, batchIndex: Int, lossValue: Double, metricValues: List<Double>)

Appends tracked data from one batch event.

appendEpoch

fun appendEpoch(epochTrainingEvent: EpochTrainingEvent)

Appends one EpochTrainingEvent.

fun appendEpoch(epochIndex: Int, lossValue: Double, metricValues: List<Double>, valLossValue: Double?, valMetricValues: List<Double>?)

Appends tracked data from one epoch event.

eventsByEpoch

fun eventsByEpoch(epochIndex: Int): TreeMap<Int, BatchTrainingEvent>?

Returns all BatchTrainingEvent of the specific epoch.

lastBatchEvent

fun lastBatchEvent(): BatchTrainingEvent

Returns last BatchTrainingEvent

lastEpochEvent

fun lastEpochEvent(): EpochTrainingEvent

Returns last EpochTrainingEvent.

Properties

batchHistory

val batchHistory: List<BatchTrainingEvent>

Batch history.

epochHistory

val epochHistory: List<EpochTrainingEvent>

Epoch history.