geney 1.2.46__py2.py3-none-any.whl → 1.2.48__py2.py3-none-any.whl

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, savepath=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,7 +88,7 @@ class SurvivalAnalysis:
             auc_vals.append(auc)
             if plot:
                 if count == 0:
-                    ax = kmf.plot_survival_function(ci_show=True, color="#2430e0", lw=2)
+                    kmf.plot_survival_function(ax=ax, ci_show=True, color="#2430e0", lw=2)
                 else:
                     kmf.plot_survival_function(ax=ax, ci_show=True, color="#e60215", lw=2)
                 count += 1
@@ -96,21 +96,24 @@ class SurvivalAnalysis:
         p_value = self.log_rank(df[df[feature] == 1], df[df[feature] == 0])
 
         if plot:
-            ax.text(0.5, 0.85, rf'$Survival p{{v}}$: {p_value:.4f}', transform=ax.transAxes, fontsize=9,
+            ax.text(0.75, 0.6, rf'Survival $p{{v}}$: {p_value:.3e}', transform=ax.transAxes, fontsize=10,
                     horizontalalignment='center')
             # 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=12)
-            ax.legend(fontsize=12, frameon=False, loc='upper right')
+            ax.tick_params(axis="both", which="major", labelsize=10)
+            legend = ax.legend(fontsize=9, 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.tight_layout()
-
-            # plt.show()
+            if savepath is not None:
+                plt.savefig(savepath, bbox_inches='tight', dpi=300)
+            plt.show()
 
         results['p_value'] = p_value
         results['auc_target'] = auc_vals[-1]

{geney-1.2.46.dist-info → geney-1.2.48.dist-info}/METADATA

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

{geney-1.2.46.dist-info → geney-1.2.48.dist-info}/RECORD

@@ -12,7 +12,7 @@ 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=usBLTwGnFzYOcl70p3lwM4SEfy1KG-UVyx-dDmXTQWY,6196
+geney/survival_utils.py,sha256=50i5oAZ_-jK9RLC3Bv2Q5IIODAilwpeBkw1fkHpj1bo,6508
 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.46.dist-info/METADATA,sha256=-MelhppfvYVO4W_1hwKuToXWF-SEubjyQLuGJ_DuxeQ,948
-geney-1.2.46.dist-info/WHEEL,sha256=fS9sRbCBHs7VFcwJLnLXN1MZRR0_TVTxvXKzOnaSFs8,110
-geney-1.2.46.dist-info/top_level.txt,sha256=O-FuNUMb5fn9dhZ-dYCgF0aZtfi1EslMstnzhc5IIVo,6
-geney-1.2.46.dist-info/RECORD,,
+geney-1.2.48.dist-info/METADATA,sha256=SKD2Rs1epKVxxTHIvT9sMHD-F0MpBznFCKwSSrfiG2Y,948
+geney-1.2.48.dist-info/WHEEL,sha256=fS9sRbCBHs7VFcwJLnLXN1MZRR0_TVTxvXKzOnaSFs8,110
+geney-1.2.48.dist-info/top_level.txt,sha256=O-FuNUMb5fn9dhZ-dYCgF0aZtfi1EslMstnzhc5IIVo,6
+geney-1.2.48.dist-info/RECORD,,