createAutoencoderModel: Function for creating a symmetric autoencoder model.

View source: R/createAutoencoderModel.R

createAutoencoderModelR Documentation

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 March 27, 2024, 4:26 a.m.