From 2296871ceb72acf9031e10edff8ff00ad7229fd6 Mon Sep 17 00:00:00 2001 From: morrisnein Date: Wed, 17 May 2023 10:41:07 +0300 Subject: [PATCH] add logging in PymfeExtractor --- .../meta_features_extractors/pymfe_extractor.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/meta_automl/data_preparation/meta_features_extractors/pymfe_extractor.py b/meta_automl/data_preparation/meta_features_extractors/pymfe_extractor.py index 36cb9d45..8dbc728f 100644 --- a/meta_automl/data_preparation/meta_features_extractors/pymfe_extractor.py +++ b/meta_automl/data_preparation/meta_features_extractors/pymfe_extractor.py @@ -3,6 +3,7 @@ from typing import List, Union, Dict, Any import pandas as pd +from golem.core.log import default_log from pymfe.mfe import MFE from meta_automl.data_preparation.dataset import DatasetCache @@ -18,6 +19,7 @@ def __init__(self, extractor_params: Dict[str, Any] = None, datasets_loader: Dat self.extractor_params = extractor_params if extractor_params is not None else self.DEFAULT_PARAMS self._datasets_loader = datasets_loader or OpenMLDatasetsLoader() self._extractor = MFE(**self.extractor_params) + self._logger = default_log(self) @property def datasets_loader(self) -> DatasetsLoader: @@ -34,6 +36,7 @@ def extract(self, datasets: List[Union[DatasetCache, str]], fill_input_nans: boo if isinstance(dataset, str): dataset = DatasetCache(dataset) + self._logger.info(f'Extracting meta features of the dataset {dataset.name}...') if (use_cached and (mfs := self._get_meta_features_cache(dataset.name, meta_feature_names))): meta_features[dataset.name] = mfs