rpudlPretrain {rpud} | R Documentation |
Pre-train a RPUDL
deep learning model.
rpudlPretrain(
model,
learning.rate = 0,
batch.size = 128,
iterations = 128,
display = 100,
seed = 0,
...,
log.level = 1
)
model |
a |
learning.rate |
manual override of the learning rate, ignored if zero or negative |
batch.size |
the size of a data batch for each training iteration |
iterations |
maximum number of training iterations |
display |
a logical to display training progress |
seed |
seed of random number generation |
... |
further arguments to be passed to or from methods |
log.level |
level of the method output (0 = silent, 1 = basic output, 2 = detailed output) |
Only the fully-connected layers will be pre-trained via one of the following mechanisms:
Masked autoencoder
Gaussian autoencoder
Restricted Boltzmann Machine
An rpudl
object with pre-trained parameters.
Chi Yau
chi.yau@r-tutor.com
rpudl,
rpudlTrain,
rpudlPredict,
## Not run:
# create data source
ds <- rpudlCreateDataSource(
data.format="lmdb",
data.dir="data/mnist",
train.data="mnist-official-data_train_lmdb",
test.data="mnist-official-data_test_lmdb",
data.shape=c(28, 28)
)
# create model
model <- rpudl(
"masked_autoencoder.prototxt",
data.source=ds
)
# pre-train
model <- rpudlPretrain(model, batch=100, iter=1000)
## End(Not run)