build_hidden_encoder: Build the encoder for a VAE

View source: R/build_hidden_encoder.R

build_hidden_encoderR Documentation

Build the encoder for a VAE

Description

Build the encoder for a VAE

Usage

build_hidden_encoder(
  input_size,
  layers,
  activations = rep("sigmoid", length(layers))
)

Arguments

input_size

an integer representing the number of items

layers

a list of integers giving the size of each hidden layer

activations

a list of strings, the same length as layers

Value

two tensors: the input layer to the VAE and the last hidden layer of the encoder


ML2Pvae documentation built on May 23, 2022, 9:05 a.m.