알고리즘을 실행하는 데 필요한 패키지를 불러옵니다.
[in :]
import pandas as pd
import numpy as np
import pyfpgrowth as fp
입력 데이터를 준비합니다.
[in :]
dict1 = {
'id':[0, 1, 2, 3],
'items':[["wickets", "pads"],
["bat", "wickets", "pads", "helmet"],
["helmet", "pads"],
["bat", "pads", "helmet"]]
}
transactionSet = pd.DataFrame(dict1)
find_frequent_patterns() 함수에 파라미터를 전달하여 규칙을 생성합니다. 이 함수에 입력하는 두 번째 파라미터인 1은 패턴이 만족해야 할 최소 빈도를 의미합니다.
[in :]
patterns = fp.find_frequent_patterns(transactionSet['items'], 1)