Update association_rules.py

rasbt · Jul 2, 2024 · aad9473 · aad9473
1 parent f313efe
commit aad9473
Showing 1 changed file with 22 additions and 15 deletions.
diff --git a/mlxtend/frequent_patterns/association_rules.py b/mlxtend/frequent_patterns/association_rules.py
@@ -13,21 +13,28 @@
 import numpy as np
 import pandas as pd
 
-_metrics = ["antecedent support",
-            "consequent support",
-            "support",
-            "confidence",
-            "lift",
-            "leverage",
-            "conviction",
-            "zhangs_metric",
-            "jaccard",
-            "certainty",
-            "kulczynski"
-            ]
-
-
-def association_rules(df: pd.DataFrame, metric="confidence", min_threshold=0.8, support_only=False, return_metrics: list = _metrics) -> pd.DataFrame:
+_metrics = [
+    "antecedent support",
+    "consequent support",
+    "support",
+    "confidence",
+    "lift",
+    "leverage",
+    "conviction",
+    "zhangs_metric",
+    "jaccard",
+    "certainty",
+    "kulczynski",
+]
+
+
+def association_rules(
+    df: pd.DataFrame,
+    metric="confidence",
+    min_threshold=0.8,
+    support_only=False,
+    return_metrics: list = _metrics,
+) -> pd.DataFrame:
     """Generates a DataFrame of association rules including the
     metrics 'score', 'confidence', and 'lift'