40 Algorithms Every Programmer Should Know

Code for using FP-growth

Let's see how we can generate association rules using the FP-growth algorithm in Python. For this, we will be using the pyfpgrowth package. First, if we have never used pyfpgrowth before, let's install it first:

!pip install pyfpgrowth

Then, let's import the packages that we need to use to implement this algorithm:

import pandas as pdimport numpy as npimport pyfpgrowth as fp

Now we will create the input data in the form of transactionSet:

dict1 = { 'id':[0,1,2,3], 'items':[["wickets","pads"], ["bat","wickets","pads","helmet"], ["helmet","pad"], ["bat","pads","helmet"]] }transactionSet = pd.DataFrame(dict1)

Once the input data is generated, we will generate patterns that will be based on the parameters that we passed ...