From 0a84a4b4dc0d5d89ec57b93737044591282515b8 Mon Sep 17 00:00:00 2001
From: Tanushree Tunstall <tanu@tunstall.in>
Date: Fri, 27 May 2022 14:32:41 +0100
Subject: [PATCH] update

---
 UQ_or_impute.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/UQ_or_impute.py b/UQ_or_impute.py
index 0c78bc2..05f6d5e 100644
--- a/UQ_or_impute.py
+++ b/UQ_or_impute.py
@@ -39,7 +39,7 @@ my_df3 = pd.DataFrame(KNN(n_neighbors= 5, weights="uniform").fit_transform(df[or
 my_df3.columns = or_cols
 
 imputer = KNNImputer(n_neighbors=2, weights="uniform")
-my_df4  = imputer.fit_transform(my_df2)
+my_df4  = pd.DataFrame(imputer.fit_transform(df[or_cols]), columns = or_cols)
 
 #all(my_df3==my_df4)
 my_df3['log10_or_mychisq'].value_counts().plot(kind = 'hist')