Coverage for mindsdb / integrations / handlers / autosklearn_handler / config.py: 0%

1import sys 

2from dataclasses import dataclass 

3from autosklearn.metrics import Scorer 

4 

5 

6@dataclass(frozen=True) 

7class BaseConfig: 

8 time_left_for_this_task: int = 3600 

9 per_run_time_limit: int = None 

10 n_jobs: int = None 

11 metric: Scorer = 'accuracy' 

12 ensemble_size: int = None 

13 ensemble_nbest: int = 50 

14 initial_configurations_via_metalearning: int = 25 

15 resampling_strategy: str = 'holdout' 

16 

17 def __post_init__(self): 

18 object.__setattr__(self, 'metric', getattr(sys.modules[__name__], self.metric)) 

19 

20 

21@dataclass(frozen=True) 

22class ClassificationConfig(BaseConfig): 

23 pass 

24 

25 

26@dataclass(frozen=True) 

27class RegressionConfig(BaseConfig): 

28 pass