|
template<typename T > |
static void | callMethod (std::shared_ptr< T > callback, const std::string &methodName, Model &model) |
| Calls the method of the callback with the given logs.
|
|
|
static void | checkMetric (const std::string &metric, const std::vector< std::string > &metrics) |
|
static std::unordered_map< std::string, Logs > | getLogs (Model &model) |
|
◆ callMethod()
template<typename T >
static void NeuralNet::Callback::callMethod |
( |
std::shared_ptr< T > | callback, |
|
|
const std::string & | methodName, |
|
|
Model & | model ) |
|
inlinestatic |
Calls the method of the callback with the given logs.
- Template Parameters
-
T | The type of the callback |
- Parameters
-
callback | A shared_ptr to the callback |
methodName | The name of the method to call (onTrainBegin, onTrainEnd, onEpochBegin, onEpochEnd, onBatchBegin, onBatchEnd) |
logs | The logs to pass to the method |
- Warning
- There should be consistency between the names of the logs and the metrics of the callbacks
◆ onEpochEnd()
virtual void NeuralNet::Callback::onEpochEnd |
( |
Model & | model | ) |
|
|
pure virtual |
◆ onTrainBegin()
virtual void NeuralNet::Callback::onTrainBegin |
( |
Model & | model | ) |
|
|
pure virtual |
◆ onTrainEnd()
virtual void NeuralNet::Callback::onTrainEnd |
( |
Model & | model | ) |
|
|
pure virtual |
The documentation for this class was generated from the following file: