ai.djl.basicmodelzoo.basic

Class Mlp

java.lang.Object

ai.djl.nn.AbstractBlock

ai.djl.nn.SequentialBlock

ai.djl.basicmodelzoo.basic.Mlp

All Implemented Interfaces:

ai.djl.nn.Block

public class Mlp
extends ai.djl.nn.SequentialBlock

Multilayer Perceptron (MLP) NeuralNetworks.

A multilayer perceptron (MLP) is a feedforward artificial neural network that generates a set of outputs from a set of inputs. An MLP is characterized by several layers of input nodes connected as a directed graph between the input and output layers. MLP uses backpropogation for training the network.

MLP is widely used for solving problems that require supervised learning as well as research into computational neuroscience and parallel distributed processing. Applications include speech recognition, image recognition and machine translation.

Field Summary

Fields inherited from class ai.djl.nn.AbstractBlock

inputNames, inputShapes

Constructor Summary

Constructors

Constructor and Description
Mlp(int input, int output, int[] hidden) Create an MLP NeuralNetwork using RELU.
Mlp(int input, int output, int[] hidden, java.util.function.Function<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> activation) Create an MLP NeuralNetwork.

Method Summary

Methods inherited from class ai.djl.nn.SequentialBlock

add, add, addAll, addAll, forward, getChildren, getDirectParameters, getOutputShapes, getParameterShape, initialize, loadParameters, removeLastBlock, replaceLastBlock, saveParameters, toString

Methods inherited from class ai.djl.nn.AbstractBlock

beforeInitialize, cast, clear, describeInput, getParameters, isInitialized, readInputShapes, saveInputShapes, setInitializer, setInitializer

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface ai.djl.nn.Block

forward, validateLayout

Constructor Detail

Mlp

public Mlp(int input,
           int output,
           int[] hidden)

Create an MLP NeuralNetwork using RELU.

Parameters:

input - the size of the input vector

output - the size of the input vector

hidden - the sizes of all of the hidden layers

Mlp

public Mlp(int input,
           int output,
           int[] hidden,
           java.util.function.Function<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> activation)

Create an MLP NeuralNetwork.

Parameters:

input - the size of the input vector

output - the size of the input vector

hidden - the sizes of all of the hidden layers

activation - the activation function to use