Coverage for addmo/s3_model_tuning/scoring/validator_factory.py: 75%

1from addmo.s3_model_tuning.scoring.abstract_scorer import ValidationScoring 

2from addmo.s3_model_tuning.scoring.validator import CrossValidation 

3from addmo.s3_model_tuning.scoring.validator import NoValidation 

4from addmo.s3_model_tuning.config.model_tuning_config import ModelTunerConfig 

5 

6class ValidatorFactory: 

7 """ 

8 Factory for creating validator instances. 

9 """ 

10 

11 @staticmethod 

12 def ValidatorFactory(config: ModelTunerConfig) -> ValidationScoring: 

13 if config.validation_score_mechanism == 'cv': 

14 return CrossValidation(config) 

15 if config.validation_score_mechanism == 'none': 

16 return NoValidation(config) 

17 # Add more conditions for other scoring mechanisms or get them dynamically like in the 

18 # other factory methods 

19 else: 

20 raise ValueError("Unknown validation scoring mechanism: " 

21 f"{config.config_model_tuner.validation_score_mechanism}. " 

22 "Available validation scoring mechanisms are: cv")