Coverage for mlair/workflows/custom_workflows.py: 0%

1"""Collection of custom workflows.""" 

2 

3__author__ = "Lukas Leufen" 

4__date__ = '2022-07-20' 

5 

6 

7from mlair.run_modules import ExperimentSetup, PreProcessing, ModelSetup, Training 

8from mlair.workflows.default_workflow import DefaultWorkflow 

9 

10 

11class HyperSearchWorkflow(DefaultWorkflow): 

12 """ 

13 A workflow executing ExperimentSetup, PreProcessing, ModelSetup, and Training. PostProcessing is not added to this 

14 workflow which is intended to be used in a training cycle. 

15 """ 

16 

17 def _setup(self, **kwargs): 

18 """Set up default workflow.""" 

19 self.add(ExperimentSetup, **kwargs) 

20 self.add(PreProcessing) 

21 self.add(ModelSetup) 

22 self.add(Training)