add threshold on correlation values

utils.py

@@ -11,13 +11,13 @@ def clean_csv_file(csv_file):
     return df
 
 
-def build_graph(csv_file):
+def build_graph(csv_file, threshold):
     features = clean_csv_file(csv_file)
     links = features.corr(method='kendall').fillna(0).stack().reset_index()
     links.columns = ['var1', 'var2', 'value']
 
     # Keep only correlation over a threshold
-    links_filtered = links.loc[(links['value'] > 0.1) & (links['var1'] != links['var2'])]
+    links_filtered = links.loc[(links['value'] > threshold) & (links['var1'] != links['var2'])]
     iter_values = iter(set(links_filtered['value']))
 
     G = nx.from_pandas_edgelist(links_filtered, 'var1', 'var2')
@@ -46,9 +46,9 @@ class MplColorHelper:
         return f"rgb({r},{g},{b})"
     
 
-def display_graph(csv_file):
+def display_graph(csv_file, threshold):
 
-    G = build_graph(csv_file)
+    G = build_graph(csv_file, threshold=threshold)
 
     CM_NAME = "Wistia"