Stores hyperparameter-search progress so interrupted model training can be
resumed from an analysis directory.
Value
An R6 class generator for persistent training caches.
Methods
TrainingCache$new()
Creates a new training cache
Arguments
inDirPath to the analysis directory
TrainingCache$isParamGridIdentical()
Checks whether the parameter grid in the model settings is identical to
the cached parameters.
Usage
TrainingCache$isParamGridIdentical(inModelParams)
Arguments
inModelParamsParameter grid from the model settings
Returns
Whether the provided and cached parameter grid is identical
TrainingCache$saveGridSearchPredictions()
Saves the grid search results to the training cache
Usage
TrainingCache$saveGridSearchPredictions(inGridSearchPredictions)
Arguments
inGridSearchPredictionsGrid search predictions
TrainingCache$saveModelParams()
Saves the parameter grid to the training cache
Usage
TrainingCache$saveModelParams(inModelParams)
Arguments
inModelParamsParameter grid from the model settings
TrainingCache$getGridSearchPredictions()
Gets the grid search results from the training cache
Usage
TrainingCache$getGridSearchPredictions()
Returns
Grid search results from the training cache
TrainingCache$isFull()
Check if cache is full
TrainingCache$getLastGridSearchIndex()
Gets the last index from the cached grid search
Usage
TrainingCache$getLastGridSearchIndex()
Returns
Last grid search index
TrainingCache$dropCache()
Remove the training cache from the analysis path
Usage
TrainingCache$dropCache()
Trims the performance of the hyperparameter results by removing
the predictions from all but the best performing hyperparameter
Usage
TrainingCache$trimPerformance(hyperparameterResults)
Arguments
hyperparameterResultsList of hyperparameter results
TrainingCache$clone()
The objects of this class are cloneable with this method.
Usage
TrainingCache$clone(deep = FALSE)
Arguments
deepWhether to make a deep clone.
Examples
cacheDirectory <- tempfile("training-cache-")
dir.create(cacheDirectory)
cache <- trainingCache$new(cacheDirectory)
cache$saveModelParams(list(list(sizeHidden = 64)))
cache$isParamGridIdentical(list(list(sizeHidden = 64)))
#> [1] TRUE
unlink(cacheDirectory, recursive = TRUE)