From 7273eaadb6891eca5a785b7796195a62587fdbe1 Mon Sep 17 00:00:00 2001 From: GB <1106205841@qq.com> Date: Thu, 19 Sep 2024 01:26:39 +0800 Subject: [PATCH] Fix: Correct the parameters passed to the plotting function in PCA Decomposition --- geochemistrypi/data_mining/model/decomposition.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/geochemistrypi/data_mining/model/decomposition.py b/geochemistrypi/data_mining/model/decomposition.py index 95e60d7..1020d69 100644 --- a/geochemistrypi/data_mining/model/decomposition.py +++ b/geochemistrypi/data_mining/model/decomposition.py @@ -327,8 +327,10 @@ def special_components(self, **kwargs: Union[Dict, np.ndarray, int]) -> None: GEOPI_OUTPUT_ARTIFACTS_IMAGE_MODEL_OUTPUT_PATH = os.getenv("GEOPI_OUTPUT_ARTIFACTS_IMAGE_MODEL_OUTPUT_PATH") # Draw graphs when the number of principal components > 3 if kwargs["components_num"] > 3: + # choose two of dimensions to draw - two_dimen_axis_index, two_dimen_pc_data = self.choose_dimension_data(self.pc_data, 2) + two_dimen_axis_index, two_dimen_pc_data = self.choose_dimension_data(self.X_reduced, 2) + two_dimen_reduced_data = self.X_reduced.iloc[:, two_dimen_axis_index] self._biplot( reduced_data=two_dimen_reduced_data, @@ -340,7 +342,7 @@ def special_components(self, **kwargs: Union[Dict, np.ndarray, int]) -> None: ) # choose three of dimensions to draw - three_dimen_axis_index, three_dimen_pc_data = self.choose_dimension_data(self.pc_data, 3) + three_dimen_axis_index, three_dimen_pc_data = self.choose_dimension_data(self.X_reduced, 3) three_dimen_reduced_data = self.X_reduced.iloc[:, three_dimen_axis_index] self._triplot( reduced_data=three_dimen_reduced_data, @@ -352,7 +354,7 @@ def special_components(self, **kwargs: Union[Dict, np.ndarray, int]) -> None: ) elif kwargs["components_num"] == 3: # choose two of dimensions to draw - two_dimen_axis_index, two_dimen_pc_data = self.choose_dimension_data(self.pc_data, 2) + two_dimen_axis_index, two_dimen_pc_data = self.choose_dimension_data(self.X_reduced, 2) two_dimen_reduced_data = self.X_reduced.iloc[:, two_dimen_axis_index] self._biplot( reduced_data=two_dimen_reduced_data,