First implement changes in `target_encoder` and `categorical_data_processor` before starting this. - [x] Add parameter `model_type` which can be [`classification`, `regression`] to the input. https://github.com/PythonPredictions/cobra/blob/6ef6723440add755ec94b46ad87d1a913d027776/cobra/preprocessing/preprocessor.py#L56 - [x] Make sure to change the `from_params`: https://github.com/PythonPredictions/cobra/blob/6ef6723440add755ec94b46ad87d1a913d027776/cobra/preprocessing/preprocessor.py#L68 - [x] Make sure that the `pipeline` has also this new parameter when saving (`pipeline["model_type"] = 'classification'`), loading and validating: https://github.com/PythonPredictions/cobra/blob/48d8a9aefbcbde7ecc898da69974a9b23727744c/cobra/preprocessing/preprocessor.py#L396 and https://github.com/PythonPredictions/cobra/blob/48d8a9aefbcbde7ecc898da69974a9b23727744c/cobra/preprocessing/preprocessor.py#L424 - [x] Document and write/modify unit tests
First implement changes in
target_encoderandcategorical_data_processorbefore starting this.model_typewhich can be [classification,regression] to the input.cobra/cobra/preprocessing/preprocessor.py
Line 56 in 6ef6723
from_params:cobra/cobra/preprocessing/preprocessor.py
Line 68 in 6ef6723
pipelinehas also this new parameter when saving (pipeline["model_type"] = 'classification'), loading and validating:cobra/cobra/preprocessing/preprocessor.py
Line 396 in 48d8a9a
cobra/cobra/preprocessing/preprocessor.py
Line 424 in 48d8a9a