Fixing memory usage increase

zazass8 · zazass8 · commit 2b0aa6b677e0 · 2025-01-25T22:37:17.000+01:00
diff --git a/mlxtend/frequent_patterns/fpcommon.py b/mlxtend/frequent_patterns/fpcommon.py
@@ -31,7 +31,7 @@ def setup_fptree(df, min_support):
     )
 
     item_support = np.array(
-        np.sum(np.logical_or(df.values == 1, df.values is True), axis=0)
+        np.nansum(df.values, axis=0)
         / (float(num_itemsets) - np.nansum(disabled, axis=0))
     )
     item_support = item_support.reshape(-1)

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ def setup_fptree(df, min_support):`
`31`	`31`	`)`
`32`	`32`
`33`	`33`	`item_support = np.array(`
`34`		`- np.sum(np.logical_or(df.values == 1, df.values is True), axis=0)`
	`34`	`+ np.nansum(df.values, axis=0)`
`35`	`35`	`/ (float(num_itemsets) - np.nansum(disabled, axis=0))`
`36`	`36`	`)`
`37`	`37`	`item_support = item_support.reshape(-1)`