finol.model_layer.ModelInstantiator

class finol.model_layer.ModelInstantiator(load_dataset_output)

Class to instantiate model class based on the provided configuration.

Parameters:

load_dataset_output (Dict) – Dictionary containing output from function load_dataset().

Methods

instantiate_model([sampled_params])

Instantiate a model based on the configuration provided in the class.

instantiate_model(sampled_params=None)

Instantiate a model based on the configuration provided in the class.

This method retrieves the model class defined in the configuration, and creates an instance of the model that is moved to the designated device.

Returns:

An instance of the specified model class.

Return type:

Module