Fix: Correct the parameters passed to the plotting function in PCA De…

…composition
ZJUEarthData · Sep 18, 2024 · 7273eaa · 7273eaa
1 parent c4c4bca
commit 7273eaa
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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,