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 data from configuration file
Prints basic information from the model configuration
- get_evaluate(data: DataModule) SODAeval [source]#
- get_model(data: DataModule) Model [source]#
- get_plotter(data: DataModule) Plotter [source]#
- get_progress_board() ProgressBoard [source]#
- get_test_dataset() DataModule [source]#
- get_train_dataset() DataModule [source]#