Coverage for addmo/s2_data_tuning/config/data_tuning_config.py: 100%

1import os 

2from pydantic import BaseModel, Field 

3from addmo.util.load_save_utils import root_dir 

4from pydantic import conlist 

5 

6class DataTuningFixedConfig(BaseModel): 

7 abs_path_to_data: str = Field( 

8 os.path.join(root_dir(),'addmo_examples','raw_input_data','InputData.xlsx'), 

9 description="Absolute path to raw system_data", 

10 ) 

11 name_of_raw_data: str = Field( 

12 "test_raw_data", description="Name of the raw system_data set" 

13 ) 

14 name_of_tuning: str = Field( 

15 "data_tuning_experiment_fixed", description="Name of the system_data tuning configuration" 

16 ) 

17 name_of_target: str = Field("FreshAir Temperature", description="Output of prediction") 

18 features: list[str] = Field( 

19 [ 

20 "Schedule", 

21 "Total active power", 

22 "FreshAir Temperature___diff", 

23 "FreshAir Temperature___lag1", 

24 "FreshAir Temperature___squared", 

25 "Total active power___lag1", 

26 ], 

27 description="List of features which the tuning shall result in.", 

28 

29 )