ertk.tensorflow.models.get_tf_model

ertk.tensorflow.models.get_tf_model(name: str, **kwargs) → Model

Get a TensorFlow model by name.

Parameters:

name: str

The model name. This should be the name of the module that implements the model.

**kwargs:

Keyword arguments to provide to the model creation function.

Returns:

model: Model

The uncompiled TensorFlow model.