You are on page 1of 1

import numpy as np

import matplotlib.pyplot as plt


import pandas as pd
from mlxtend.preprocessing import Transaction Encoder
from mlxtend.frequent_patterns import apriori, fpmax fpgrowth

ds = pd.read_csv('fertility.csv', header=None)
dataset = []
for i in range(lends)):
dataset.append(ds.loc[i].dropna().to_list())

te = TransactionEncoder()
te_ary = te.fit(dataset).transfrom(dataset)
df = pd.DataFrame(te_ary, columns=te.columns_)

frequent_itemsets = fpgrowth(df, min_support = 0.1, use_columns = True)


frquent_itemsets
print(frequent_itemsets)

from mlxtend.frequent_patterns import association_rules

association_rules(frequent_itemsets, metric ="confidence", min_treshold=0.7 )


rules = association_rules(frequent_itemsets, metric = "confidence", mintreshold=0)
print(rules)

You might also like