ertk.tensorflow.models.get_tf_model_fn

ertk.tensorflow.models.get_tf_model_fn(name: str, **kwargs) → Callable[[...], Model | Pipeline]

Get a TensorFlow model creation function.

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_fn: callable

A method that takes arguments and returns a Model instance.