ertk.tensorflow.models

TensorFlow implementations of various models.

get_tf_model(name, **kwargs)

Get a TensorFlow model by name.

get_tf_model_fn(name, **kwargs)

Get a TensorFlow model creation function.

TFModelConfig(optimiser, opt_params, ...)

Configuration for a TensorFlow model.

Models

aldeneh2017

Implementation of the model architecture from [Re7c0ef8cdfca-1].

audeep

Implementation of auDeep model from [R80f428b8fb55-1].

depinto2020

Implementation of short model from [Rbdcdae44a78f-1].

iskhakova2020

Implementation of model from [Recf2e1b508d8-1].

latif2019

Implementation of the model from [R34cc87fda321-1].

mlp

Basic multi-layer perceptron (MLP) model.

rbm

Implementation of Restricted Boltzmann Machines (RBMs) [R161231ffb332-1] in TensorFlow.

zhang2019

Implementation of the model from [R36f577c73fe8-1].

zhao2019

Implementation of the model from [R9f110c8cdc04-1].

Layers

layers

Custom Keras layers for use in TensorFlow models.