Simd Library Documentation.

Home | Release Notes | Download | Documentation | Forum | SourceForge | GitHub
TrainOptions Struct Reference

Contains a set of training options. More...

Public Types

enum  InitType { Xavier }
 
enum  LossType { Mse }
 
enum  UpdateType { AdaptiveGradient }
 

Public Member Functions

 TrainOptions ()
 Default constructor.
 

Data Fields

InitType initType
 Method to initialize weights.
 
LossType lossType
 Loss function type.
 
UpdateType updateType
 Weights' update type.
 
size_t threadNumber
 Number of threads used to train. Use -1 to auto detect thread number.
 
size_t epochStart
 Start epoch. It is used to continue training process.
 
size_t epochFinish
 Finish epoch. Describes total epoch number.
 
size_t batchSize
 A batch size.
 
float alpha
 Describes training speed.
 
float epsilon
 Used to prevent division by zero.
 

Detailed Description

Contains a set of training options.