createAutoencoderModel: Function for creating a symmetric autoencoder model.

In ANTsX/ANTsRNet: Neural Networks for Medical Image Processing

Function for creating a symmetric autoencoder model.

Description

Builds an autoencoder based on the specified array definining the number of units in the encoding branch. Ported to Keras R from the Keras python implementation here:

Usage

createAutoencoderModel(
  numberOfUnitsPerLayer,
  activation = "relu",
  initializer = "glorot_uniform"
)

Arguments

numberOfUnitsPerLayer	vector defining the number of units in the encoding branch
activation	activation type for the dense layers
initializer	initializer type for the dense layers

Details

https://github.com/XifengGuo/DEC-keras

Value

two models: the encoder and auto-encoder

Author(s)

Tustison NJ

Examples

library( ANTsRNet )
library( keras )

ae <- createAutoencoderModel( c( 784, 500, 500, 2000, 10 ) )

ANTsX/ANTsRNet documentation built on Nov. 21, 2024, 4:07 a.m.