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.