diff options
| author | Alexander Kabui | 2022-03-18 14:57:25 +0300 |
|---|---|---|
| committer | Alexander Kabui | 2022-03-18 14:57:25 +0300 |
| commit | 0964e2563a96130e306edd31cd91fee4b378697d (patch) | |
| tree | d7002d01fb3398f9cc22a82667f74a6e5699cdb8 | |
| parent | b158f11588afc4c93c5daf17ed0195f392e2e2d7 (diff) | |
| download | genenetwork2-0964e2563a96130e306edd31cd91fee4b378697d.tar.gz | |
code to fetch scree data from gn3
| -rw-r--r-- | wqflask/wqflask/correlation_matrix/show_corr_matrix.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/wqflask/wqflask/correlation_matrix/show_corr_matrix.py b/wqflask/wqflask/correlation_matrix/show_corr_matrix.py index 88d62045..cf37b9e9 100644 --- a/wqflask/wqflask/correlation_matrix/show_corr_matrix.py +++ b/wqflask/wqflask/correlation_matrix/show_corr_matrix.py @@ -37,6 +37,7 @@ from gn3.computations.pca import compute_pca from gn3.computations.pca import process_factor_loadings_tdata from gn3.computations.pca import generate_pca_temp_traits from gn3.computations.pca import cache_pca_dataset +from gn3.computations.pca import generate_scree_plot_data class CorrelationMatrix: @@ -169,6 +170,7 @@ class CorrelationMatrix: pca = self.calculate_pca() self.loadings_array = process_factor_loadings_tdata( factor_loadings=self.loadings, traits_num=len(self.trait_list)) + else: self.pca_works = "False" except: @@ -176,6 +178,7 @@ class CorrelationMatrix: self.js_data = dict(traits=[trait.name for trait in self.traits], groups=groups, + scree_data = self.scree_data, cols=list(range(len(self.traits))), rows=list(range(len(self.traits))), samples=self.all_sample_list, @@ -187,6 +190,7 @@ class CorrelationMatrix: self.loadings = pca["components"] self.scores = pca["scores"] + self.pca_obj = pca["pca"] this_group_name = self.trait_list[0][1].group.name temp_dataset = create_dataset( @@ -205,6 +209,13 @@ class CorrelationMatrix: self.pca_trait_ids = list(pca_temp_traits.keys()) + x_coord, y_coord = generate_scree_plot_data( + list(self.pca_obj.explained_variance_ratio_)) + + self.scree_data = { + "x_coord": x_coord, + "y_coord": y_coord + } return pca |