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, metricValue: Double)

Appends tracked data from one batch event.

appendEpoch

fun appendEpoch(epochTrainingEvent: EpochTrainingEvent)

Appends one EpochTrainingEvent.

fun appendEpoch(epochIndex: Int, lossValue: Double, metricValue: Double, valLossValue: Double?, valMetricValue: Double?)

Appends tracked data from one epoch event.

eventsByEpoch

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

Returns all BatchTrainingEvent of the specific epoch.

get

operator fun get(desiredField: KProperty1<EpochTrainingEvent, Double>): DoubleArray

Returns all values of one filed in EpochTrainingEvent for all epochs. All EpochTrainingEvent.metricValue for example.

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.