Skip to content

Instantly share code, notes, and snippets.

@CharlesRajendran

CharlesRajendran/retail_rules.py

Last active April 7, 2018 19:24
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save CharlesRajendran/4e29cc9a0abd8c2f6e173bf8b9ca810d to your computer and use it in GitHub Desktop.
Save CharlesRajendran/4e29cc9a0abd8c2f6e173bf8b9ca810d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
retail_rules.py
import pandas as pd
dataset = pd.read_csv("data.csv", header=None)
# apriori expects the input as list of list
# [ [item1, item2], [item2, item3, item4], ..... ] ]
transactions = []
for i in range(0, len(dataset)):
transactions.append([str(dataset.values[i, j]) for j in range(0, 20)])
from apyori import apriori
associations = apriori(transactions, min_support = 0.004, min_confidence = 0.4, min_lift = 3, min_length = 2)
print(list(associations))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment