Update reduce.py

abhi-glitchhg · Oct 30, 2023 · 288f776 · 288f776
1 parent ee2d61f
commit 288f776
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/corrfeatred/reduce.py b/corrfeatred/reduce.py
@@ -17,7 +17,7 @@ def reduce_features(correlation_matrix, threshhold=0.75,policy='min', random_sta
 
     if random_state!=None:
         random_gen = random.Random(random_state)
-    corrmatrix = correlation_matrix.copy()
+    corrmatrix = correlation_matrix.abs().copy()
     inf_ = 4*(corrmatrix.shape[0]) + 10 # adding 10 for no reason, this could be any positive number;
     corr_matrix = corrmatrix > threshhold
     assert policy in ('min', 'max'), "wrong input parameter"