diff --git a/atomsci/ddm/pipeline/model_wrapper.py b/atomsci/ddm/pipeline/model_wrapper.py index 06dd7446..d3c59b22 100644 --- a/atomsci/ddm/pipeline/model_wrapper.py +++ b/atomsci/ddm/pipeline/model_wrapper.py @@ -338,6 +338,8 @@ def _create_output_transformers(self, dataset): # TODO: Just a warning, we may have response transformers for classification datasets in the future if self.params.prediction_type=='regression' and self.params.transformers is True: return [trans.NormalizationTransformerMissingData(transform_y=True, dataset=dataset)] + else: + return [] # **************************************************************************************** @@ -1612,7 +1614,9 @@ def _create_output_transformers(self, dataset): """ # TODO: Just a warning, we may have response transformers for classification datasets in the future if self.params.prediction_type=='regression' and self.params.transformers is True: - self.transformers = [trans.NormalizationTransformerHybrid(transform_y=True, dataset=dataset)] + return [trans.NormalizationTransformerHybrid(transform_y=True, dataset=dataset)] + else: + return [] # **************************************************************************************** class ForestModelWrapper(ModelWrapper):