| Modifier and Type | Method and Description |
|---|---|
RBM.RBMBuilder |
batchParallelism(int numThreads) |
RBM |
build() |
static RBM.RBMBuilder |
create(ActivationFunction activation,
int... layer)
Creates a new
RBM.RBMBuilder from an activation function and
layersizes. |
RBM.RBMBuilder |
lambda(double lambda)
Sets the regularization parameter lambda, defaults to 0 if not set.
|
RBM.RBMBuilder |
miniBatchSize(int size) |
RBM.RBMBuilder |
stochastic()
Sets the training mode to stochastic.
|
RBM.RBMBuilder |
stochastic(boolean stochastic)
If verbose is true, stochastic training will be used.
|
RBM.RBMBuilder |
trainingType(TrainingType type)
Sets the training type, it defaults to CPU- so only use if you want to
use the GPU.
|
RBM.RBMBuilder |
verbose()
Sets verbose to true.
|
RBM.RBMBuilder |
verbose(boolean verbose)
If verbose is true, progress indicators will be printed to STDOUT.
|
public RBM.RBMBuilder trainingType(TrainingType type)
public RBM.RBMBuilder lambda(double lambda)
public RBM.RBMBuilder miniBatchSize(int size)
size - the minibatch size to use. Batches are calculated in parallel
on every cpu core if not overridden by
batchParallelism(int).public RBM.RBMBuilder batchParallelism(int numThreads)
numThreads - set the number of threads where batches should be
calculated in parallel.public RBM.RBMBuilder verbose()
public RBM.RBMBuilder stochastic()
public RBM.RBMBuilder stochastic(boolean stochastic)
public RBM.RBMBuilder verbose(boolean verbose)
public static RBM.RBMBuilder create(ActivationFunction activation, int... layer)
RBM.RBMBuilder from an activation function and
layersizes.activation - the activation function.layer - an array of hidden layer sizes.Copyright © 2016. All rights reserved.