From 5ab52154a5413f4e3c898e07aed924e34fbe75ae Mon Sep 17 00:00:00 2001 From: hanna julienne <hanna.julienne@pasteur.fr> Date: Mon, 26 Mar 2018 17:27:22 +0200 Subject: [PATCH] adding main file --- impute_jass/impute_jass/imputation_launcher.py | 12 ++++++++++++ impute_jass/impute_jass/main.py | 0 impute_jass/impute_jass/stat_models.py | 2 +- impute_jass/setup.py | 12 ++++++------ 4 files changed, 19 insertions(+), 7 deletions(-) create mode 100644 impute_jass/impute_jass/main.py diff --git a/impute_jass/impute_jass/imputation_launcher.py b/impute_jass/impute_jass/imputation_launcher.py index e4ff662..97f487c 100644 --- a/impute_jass/impute_jass/imputation_launcher.py +++ b/impute_jass/impute_jass/imputation_launcher.py @@ -6,6 +6,18 @@ import glob import pandas as pd from .windows import ld_region_centered_window_imputation, impg_like_imputation, realigned_zfiles_on_panel + +def chromosome_imputation_entry_point(chrom, zscore, ref_panel, ld_folder, window_size=10000, imputation_style="online", buf=2500, lamb= 0.01, pinv_rcond = 0.01): + """ + Function whose only purpose is to allow the calling of the ImputationLauncher.chromosome_imputation method + from an entry point + """ + imputer = ImputationLauncher( window_size=10000, imputation_style="online", buf=2500, lamb= 0.01, pinv_rcond = 0.01) + imputer.chromosome_imputation(chrom, zscore, ref_panel, ld_folder) + + + + class ImputationLauncher(object): """ Class perform imputation of snp from summary statistic diff --git a/impute_jass/impute_jass/main.py b/impute_jass/impute_jass/main.py new file mode 100644 index 0000000..e69de29 diff --git a/impute_jass/impute_jass/stat_models.py b/impute_jass/impute_jass/stat_models.py index a9e9ea0..a5c5343 100644 --- a/impute_jass/impute_jass/stat_models.py +++ b/impute_jass/impute_jass/stat_models.py @@ -28,7 +28,7 @@ def impg_model(zt, sig_t, sig_i_t, lamb=0.01, rcond=0.01, batch=True): """ sig_t = sig_t.values np.fill_diagonal(sig_t, (1+lamb)) - sig_t_inv = sc.linalg.pinv(sig_t)#, rcond=rcond) + sig_t_inv = sc.linalg.pinv(sig_t, rcond=rcond) if batch: condition_number = np.array([np.linalg.cond(sig_t)]*sig_i_t.shape[0]) diff --git a/impute_jass/setup.py b/impute_jass/setup.py index 89c7c36..4ceebab 100644 --- a/impute_jass/setup.py +++ b/impute_jass/setup.py @@ -12,11 +12,11 @@ setup(name='impute_jass', package_data = {'impute_jass':'./data/*.csv'}, zip_safe=False - entry_points={ - 'console_scripts' : [ - 'impute_chromosome = ' - - ] - } + # entry_points={ + # 'console_scripts' : [ + # 'impute_chromosome = imputation_launcher:chromosome_imputation_entry_point' + # + # ] + # } ) -- GitLab