OneCompiler

Market_Basket

107

import pandas as pd
from mlxtend.frequent_patterns import apriori,association_rules
from mlxtend.preprocessing import TransactionEncoder

data = pd.read_csv('store_data.csv')
print(data.info())

data.dropna()

te = TransactionEncoder()
te_array = te.fit(data).transform(data)
df = pd.DataFrame(te_array, columns=te.columns_)

print(df.head())

//next line
freq_item = apriori(df, min_support=0.05, use_colnames=True)
print(freq_item)

rules = association_rules(freq_item, metric="confidence", min_threshold=0.5)
print(rules)