geney 1.2.45__py2.py3-none-any.whl → 1.2.47__py2.py3-none-any.whl

geney/oncosplice.py

@@ -402,7 +402,7 @@ def oncosplice(mut_id, splicing_threshold=0.5, protein_coding=True, primary_tran
                         sorted(cons_vector).index(next(x for x in sorted(cons_vector) if x >= affected_cons_scores)) / len(
                     cons_vector))
 
-            report = {} #OncospliceAnnotator(reference_transcript, transcript, mutation)
+            report = OncospliceAnnotator(reference_transcript, transcript, mutation)
             report['mut_id'] = mut_id
             report['oncosplice_score'] = affected_cons_scores
             report['percentile'] = percentile

geney/survival_utils.py

@@ -64,7 +64,7 @@ class SurvivalAnalysis:
                                  df[col].nunique() > 1]]  # and df[col].value_counts(normalize=True).min() >= 0.01]]
         return df
 
-    def kaplan_meier_analysis(self, df, control_label='CV', target_label='Epistasis', feature='group', plot=False, time_cap=False):
+    def kaplan_meier_analysis(self, df, control_label='CV', target_label='Epistasis', feature='group', plot=False, time_cap=False, ax=None):
         # Can only be performed on features with two unique values
         cap_time = df.groupby(feature).duration.max().min()
         # df['duration'] = df['duration'].clip(upper=cap_time)
@@ -88,21 +88,31 @@ class SurvivalAnalysis:
             auc_vals.append(auc)
             if plot:
                 if count == 0:
-                    ax = kmf.plot_survival_function()
+                    kmf.plot_survival_function(ax=ax, ci_show=True, color="#2430e0", lw=2)
                 else:
-                    kmf.plot_survival_function(ax=ax)
+                    kmf.plot_survival_function(ax=ax, ci_show=True, color="#e60215", lw=2)
                 count += 1
+
         p_value = self.log_rank(df[df[feature] == 1], df[df[feature] == 0])
 
         if plot:
-            ax.text(0.5, 0.85, f'p-value: {p_value:.4f}', transform=ax.transAxes, fontsize=12,
+            ax.text(0.5, 0.85, rf'Survival $p{{v}}$: {p_value:.3e}', transform=ax.transAxes, fontsize=10,
                     horizontalalignment='center')
-            plt.title('Kaplan-Meier Survival Curves')
-            plt.xlabel('Time')
+            # Grid and spines
+            ax.grid(True, which="major", linestyle="--", linewidth=0.5, color="grey", alpha=0.7)
+            ax.spines['top'].set_visible(False)
+            ax.spines['right'].set_visible(False)
+            ax.tick_params(axis="both", which="major", labelsize=10)
+            legend = ax.legend(fontsize=12, loc='upper right', frameon=True)
+            legend.get_frame().set_facecolor('white')  # Set the background color to white
+            legend.get_frame().set_edgecolor('black')  # Set the edge color to black
+            plt.xlabel('Time (years)')
             plt.ylabel('Survival Probability')
             if time_cap:
                 plt.xlim([0, cap_time])
-            plt.show()
+            plt.tight_layout()
+
+            # plt.show()
 
         results['p_value'] = p_value
         results['auc_target'] = auc_vals[-1]

{geney-1.2.45.dist-info → geney-1.2.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: geney
-Version: 1.2.45
+Version: 1.2.47
 Summary: A Python package for gene expression modeling.
 Home-page: https://github.com/nicolaslynn/geney
 Author: Nicolas Lynn

{geney-1.2.45.dist-info → geney-1.2.47.dist-info}/RECORD

@@ -6,13 +6,13 @@ geney/graphic_utils.py,sha256=oMsBpB9YeEn96gGpKh4MmtagJffWZbk-xPrIwHvkFhA,11016
 geney/gtex_utils.py,sha256=asL2lHyU5KsbWpV096vkf1Ka7hSo_RRfZqw7p5nERmE,1919
 geney/immune_utils.py,sha256=ZRni5ttrhpYBnmNr0d0ZatIbNPYs4nmQuoUO00SpsS4,5271
 geney/mutation_utils.py,sha256=C_kv2MB_L8LlhX3W2ooXjJ3uDoJ8zX1WeDtZKoBZJkI,1547
-geney/oncosplice.py,sha256=Tmhkswohr6lcagpci26yms_XyXU14V6MqsRcV1XvQaI,23444
+geney/oncosplice.py,sha256=nP8SEXvaZbU-K_pzIOKJ6BNAl3QJgidRiptm7xBzACM,23440
 geney/pangolin_utils.py,sha256=lLmnjJdJjqwWS85-1jlPLIjD2z14sWjzU87hS-8xxpQ,2873
 geney/power_utils.py,sha256=MehZFUdkJ2EFUot709yPEDxSkXmH5XevMebX2HD768A,7330
 geney/seqmat_utils.py,sha256=YV5DFLbfjXLIswPGvqK1-eEfwn9TUby0b2kewdGAKws,18372
 geney/spliceai_utils.py,sha256=gIGPC8u3J15A7EQrk2Elho5PbF9MmUUNopGGH-eEV8s,1873
 geney/splicing_utils.py,sha256=lGBNknnAdKhcJ3MqPQ5c9oz_NKcL2lcFAr78StjKa6o,16151
-geney/survival_utils.py,sha256=2CAkC2LsspicHIdrqsiPnjgvpr5KHDUfLFFqnRbPJqs,5762
+geney/survival_utils.py,sha256=S66tKOMwGSSFXb-2MYg8ZEuyDV64942VB4eFcXRFQas,6392
 geney/tcga_utils.py,sha256=D_BNHm-D_K408dlcJm3hzH2c6QNFjQsKvUcOPiQRk7g,17612
 geney/tis_utils.py,sha256=vA2ci4gNfwwQZlCjPpO5ehvL2NRVeM7lHI_VyfT-_10,8049
 geney/utils.py,sha256=EsKvBM-Nz2a3_4ZAhF4Dxd4PwT7_6YYKpxEN4LLgg10,2174
@@ -20,7 +20,7 @@ geney/translation_initiation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 geney/translation_initiation/tis_utils.py,sha256=AF3siFjuQH-Rs44EV-80zHdbxRMvN4woLFSHroWIETc,4448
 geney/translation_initiation/resources/kozak_pssm.json,sha256=pcd0Olziutq-6H3mFWDCD9cujQ_AlZO-iiOvBl82hqE,1165
 geney/translation_initiation/resources/tis_regressor_model.joblib,sha256=IXb4DUDhJ5rBDKcqMk9zE3ECTZZcdj7Jixz3KpoZ7OA,2592025
-geney-1.2.45.dist-info/METADATA,sha256=982qXOErqUhJqh574OygwVaFJia1nA7IWVYkMa--DnQ,948
-geney-1.2.45.dist-info/WHEEL,sha256=fS9sRbCBHs7VFcwJLnLXN1MZRR0_TVTxvXKzOnaSFs8,110
-geney-1.2.45.dist-info/top_level.txt,sha256=O-FuNUMb5fn9dhZ-dYCgF0aZtfi1EslMstnzhc5IIVo,6
-geney-1.2.45.dist-info/RECORD,,
+geney-1.2.47.dist-info/METADATA,sha256=fdtNRwV6GwITCTn1O3J0ytlq1wNYda0_sGJdMiNuZU4,948
+geney-1.2.47.dist-info/WHEEL,sha256=fS9sRbCBHs7VFcwJLnLXN1MZRR0_TVTxvXKzOnaSFs8,110
+geney-1.2.47.dist-info/top_level.txt,sha256=O-FuNUMb5fn9dhZ-dYCgF0aZtfi1EslMstnzhc5IIVo,6
+geney-1.2.47.dist-info/RECORD,,