From d3ecd378ce3136a5a1d163fda47f9116eb2ef3f7 Mon Sep 17 00:00:00 2001 From: GB <1106205841@qq.com> Date: Thu, 19 Sep 2024 20:57:29 +0800 Subject: [PATCH] fix: Correct the parameters passed to the plotting function in PCA Decomposition --- geochemistrypi/data_mining/model/decomposition.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/geochemistrypi/data_mining/model/decomposition.py b/geochemistrypi/data_mining/model/decomposition.py index 95e60d7..d2ee6b5 100644 --- a/geochemistrypi/data_mining/model/decomposition.py +++ b/geochemistrypi/data_mining/model/decomposition.py @@ -328,7 +328,7 @@ def special_components(self, **kwargs: Union[Dict, np.ndarray, int]) -> None: # 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 +340,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 +352,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,