fundaml.models

Module Contents

Classes

AbstractNN	An abstract base class for all neural network models.
SampleNNClassifier	A classifier model built upon a multilayer perceptron.
SampleCNN	A sample Convolutional Neural Network (CNN) model.
SampleRNN	A sample Recurrent Neural Network (RNN) model.

class fundaml.models.AbstractNN(short_name)

Bases: torch.nn.Module

An abstract base class for all neural network models.

class fundaml.models.SampleNNClassifier(short_name)

Bases: AbstractNN

A classifier model built upon a multilayer perceptron.

forward(x)

class fundaml.models.SampleCNN(short_name)

Bases: AbstractNN

A sample Convolutional Neural Network (CNN) model.

forward(x)

class fundaml.models.SampleRNN(short_name, input_size=28, hidden_size=128, num_layers=2, num_classes=10)

Bases: AbstractNN

A sample Recurrent Neural Network (RNN) model.

forward(x)