GPUMLib  0.2.2
GPU Machine Learning Library
Public Member Functions | List of all members

Represents a Deep Belief Network (Host - CPU). More...

#include <HostDBN.h>

Public Member Functions

 DBNhost (HostArray< int > &layers, HostMatrix< cudafloat > &inputs, cudafloat initialLearningRate, cudafloat momentum=DEFAULT_MOMENTUM, bool useBinaryValuesVisibleReconstruction=false, cudafloat stdWeights=STD_WEIGHTS)
 
RBMhostGetRBM (int layer)
 
int GetNumberRBMs () const
 
void Train (int epochs, int cd)
 
void Train (int epochs, int cd, cudafloat errorStop)
 

Detailed Description

Represents a Deep Belief Network (Host - CPU).

Examples:
DBNapp.cpp.

Definition at line 32 of file HostDBN.h.

Constructor & Destructor Documentation

DBNhost ( HostArray< int > &  layers,
HostMatrix< cudafloat > &  inputs,
cudafloat  initialLearningRate,
cudafloat  momentum = DEFAULT_MOMENTUM,
bool  useBinaryValuesVisibleReconstruction = false,
cudafloat  stdWeights = STD_WEIGHTS 
)
inline

Constructs a Deep Belief Network that can be trained using the CPU (Host).

Parameters
layersNumber of units in each layer.
inputsInputs of the training dataset. Each row of the matrix should contain a pattern (sample) and each column an input.
initialLearningRateInitial learning rate.
momentumMomentum
useBinaryValuesVisibleReconstructionUse binary values for the visibible layer reconstruction
stdWeightsDefines the maximum and minimum value for the weights. The weights will be initialized with a random number between -stdWeights and stdWeights.

Definition at line 44 of file HostDBN.h.

Member Function Documentation

int GetNumberRBMs ( ) const
inline

Gets the number of RBMs.

Returns
The number of RBMs that compose the DBN.
Examples:
DBNapp.cpp.

Definition at line 76 of file HostDBN.h.

RBMhost* GetRBM ( int  layer)
inline

Get an RBM that is part of the DBN.

Parameters
layerLayer to obtain.
Returns
The RBM corresponding to layer specified.
Examples:
DBNapp.cpp.

Definition at line 69 of file HostDBN.h.

void Train ( int  epochs,
int  cd 
)
inline

Train the DBN (train each RBM).

Parameters
epochsMaximum number of epochs that each RBM should be trained.
cdDefine the value of k in CD-k

Definition at line 83 of file HostDBN.h.

void Train ( int  epochs,
int  cd,
cudafloat  errorStop 
)
inline

Train the DBN (train each RBM).

Parameters
epochsMaximum number of epochs that each RBM should be trained.
cdDefine the value of k in CD-k
errorStopStop the training in each RBM when the Mean Square Error (MSE) is inferior to this value.

Definition at line 98 of file HostDBN.h.


The documentation for this class was generated from the following file: