From a37837c0ee53394e3c9c3cf533b321e0922b9300 Mon Sep 17 00:00:00 2001 From: bryantclc Date: Tue, 25 Aug 2020 11:19:47 +0800 Subject: [PATCH] fix code style bug --- .../weighted_ensemble_feature_selector.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/analysis/optimizer/weighted_ensemble_feature_selector.py b/analysis/optimizer/weighted_ensemble_feature_selector.py index 9649fc9..cdbfb94 100644 --- a/analysis/optimizer/weighted_ensemble_feature_selector.py +++ b/analysis/optimizer/weighted_ensemble_feature_selector.py @@ -26,15 +26,15 @@ from sklearn.tree import DecisionTreeRegressor from sklearn.linear_model import ElasticNet, Ridge from sklearn.tree import ExtraTreeRegressor -from multiprocessing import Process, Queue +import multiprocessing LOGGER = logging.getLogger(__name__) -class FeatureSelectorProcess(Process): +class FeatureSelectorProcess(multiprocessing.Process): """class feature selector each with multiprocessing""" def __init__(self, regressor, list_sample_x, list_sample_y, labels, index, sorted_index_queue, prediction_queue): - Process.__init__(self) + multiprocessing.Process.__init__(self) self._regressor = regressor self._list_sample_x = list_sample_x self._list_sample_y = list_sample_y @@ -76,7 +76,7 @@ class FeatureSelectorProcess(Process): self._prediction_queue.put(prediction) -class WeightedEnsembleFeatureSelector: +class WeightedEnsembleFeatureSelector(object): """class weighted ensemble feature selector""" def __init__(self): @@ -100,9 +100,10 @@ class WeightedEnsembleFeatureSelector: prediction_queue_list = [] for regressor in self._regressors: - sorted_index_queue = Queue() - prediction_queue = Queue() - fs_thread = FeatureSelectorProcess(regressor, list_sample_x, list_sample_y, labels, index, sorted_index_queue, prediction_queue) + sorted_index_queue = multiprocessing.Queue() + prediction_queue = multiprocessing.Queue() + fs_thread = FeatureSelectorProcess(regressor, list_sample_x, list_sample_y, + labels, index, sorted_index_queue, prediction_queue) fs_thread_list.append(fs_thread) sorted_index_queue_list.append(sorted_index_queue) prediction_queue_list.append(prediction_queue) -- GitLab