correct inversion R2 var

impute_jass/impute_jass/stat_models.py

@@ -60,6 +60,7 @@ def impg_model(zt, sig_t, sig_i_t, lamb=0.01, rcond=0.01,  batch=True):
     if np.any(mu > 30):
         print("ABERANT SNP SNiP")
     var_norm = var_in_boundaries(var, lamb)
-    mu = mu / np.sqrt(var_norm)
+    R2 = (1-var_norm)
+    mu = mu / np.sqrt(R2)
 
     return({"var" : var, "mu" : mu, "ld_score" : ld_score, "condition_number" : condition_number, "correct_inversion":correct_inversion })