utils.model_loader.ModelLoader#

class ModelLoader(cfg_path='config/config.yaml')[source]#

Bases: object

Generates a model and dataset based on parameters from a configuration file

For details see the source code.

Methods

get

Get data from configuration file

get_evaluate

get_model

get_params_file_name

get_plotter

get_progress_board

get_test_dataset

get_train_dataset

get_trainer

print_info

Prints basic information from the model configuration

get(name: str) Any[source]#

Get data from configuration file

Parameters:

name (str) – Parameter name

Returns:

Parameter value

Return type:

Any

Raises:

KeyError – The key was not found in the set of existing keys

get_evaluate(data: DataModule) SODAeval[source]#
get_model(data: DataModule) Model[source]#
get_params_file_name() str[source]#
get_plotter(data: DataModule) Plotter[source]#
get_progress_board() ProgressBoard[source]#
get_test_dataset() DataModule[source]#
get_train_dataset() DataModule[source]#
get_trainer()[source]#
print_info() None[source]#

Prints basic information from the model configuration