Update train.py

mortazavilab · Aug 13, 2024 · 6a282e6 · 6a282e6
1 parent 2c4f3d5
commit 6a282e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Topyfic/train.py b/Topyfic/train.py
@@ -61,7 +61,7 @@ def combine_LDA_models(self, data, single_trains=[]):
         """
         for i in range(len(single_trains)):
             gene_weights = pd.DataFrame(np.transpose(single_trains[i].top_models[0].model.components_),
-                                        columns=[f'Topic{i + 1}_R{self.random_state_range[i]}' for i in range(self.k)],
+                                        columns=[f'Topic{j + 1}_R{self.random_state_range[i]}' for j in range(self.k)],
                                         index=data.var.index.tolist())
             TopModel_lda_model = TopModel(name=f"{self.name}_{self.random_state_range[i]}",
                                           N=gene_weights.shape[1],