Skip to content

Commit 7f4810e

Browse files
Update variation.py
1 parent d8a98c6 commit 7f4810e

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

DimRed/variation/variation.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,9 @@
22

33

44
class _Variation:
5-
def __init__(self, standard_pipeline: Pipeline) -> None:
5+
def __init__(self, standard_pipeline: Pipeline, analysis_instance: Analysis) -> None:
66
self.standard_pipeline = standard_pipeline
7+
self.analysis_instance = analysis_instance
78

89
def create_variation(self, param_grid: Dict, reduction_method: Any, reduction_method_name: str) -> List:
910
grid = ParameterGrid(param_grid)
@@ -13,5 +14,7 @@ def create_variation(self, param_grid: Dict, reduction_method: Any, reduction_me
1314
var = [(reduction_method_name, reduction_method(**params))]
1415
pipeline = Pipeline(
1516
[step for step in self.standard_pipeline.steps] + var)
17+
self.analysis_instance.produce_combinations(
18+
reduction_method_name, self.standard_pipeline, pipeline)
1619
variations.append(pipeline)
1720
return variations

0 commit comments

Comments
 (0)