关联分析-Apriori法python代码注解

这篇博客介绍了Apriori算法在Python中的实现,通过创建频繁项集和生成关联规则来挖掘数据中的关联模式。文章详细阐述了算法的主要思想,并提供了Python代码示例。同时,博主还分享了关于极大频繁集、闭频繁集的概念,以及一些练习题以加深理解。
摘要由CSDN通过智能技术生成

自己的一点点领悟,可能会有点小错误,欢迎交流^_^

获得频繁项集

主要思想

这里写图片描述

python代码
def loadDataSet():
    return [[1,3,4],[2,3,5],[1,2,3,5],[2,5]]

createC1(dataSet)获得所有第一层的所有项集

这里写图片描述

def createC1(dataSet):
    C1 = []
    for transaction in dataSet:
        for item in transaction:
            if not [item] in C1:
                C1.append([item])

    C1.sort()
    return map(frozenset,C1)
#scanD是根据训练数据D,来判断Ck里面一堆的项集是否是频繁的。

def scanD(D,Ck,minSupport):
    ssCnt = {}
    for tid in D:
        for can in Ck:
            if can.issubset(tid):
                if not ssCnt.has_key(can): ssCnt[can] = 1
                else: ssCnt[can] += 1
    numItems = float(len(D))
    retList = []
    supportData = {}
    for key in ssCnt:
        support = ssCnt[key] / numItems
        if support >= minSupport:
            retList.insert(0,key)
        supportData[key] = support
    
当然,我可以为您提供关联规则aprioriPython代码。以下是一个简单的示例: ```python from itertools import combinations from collections import defaultdict def generate_candidates(freq_sets, k): candidates = set() for itemset1 in freq_sets: for itemset2 in freq_sets: union = itemset1.union(itemset2) if len(union) == k: candidates.add(union) return candidates def prune_candidates(candidates, freq_sets, k): pruned_candidates = set() for candidate in candidates: subsets = combinations(candidate, k) if all(subset in freq_sets for subset in subsets): pruned_candidates.add(candidate) return pruned_candidates def apriori(transactions, min_support=0.5, min_confidence=0.5): item_counts = defaultdict(int) freq_sets = set() for transaction in transactions: for item in transaction: item_counts[item] += 1 num_transactions = len(transactions) for item, count in item_counts.items(): support = count / num_transactions if support >= min_support: freq_sets.add(frozenset([item])) k = 2 while freq_sets: candidates = generate_candidates(freq_sets, k) freq_counts = defaultdict(int) for transaction in transactions: for candidate in candidates: if candidate.issubset(transaction): freq_counts[candidate] += 1 freq_sets = set() for candidate, count in freq_counts.items(): support = count / num_transactions if support >= min_support: freq_sets.add(candidate) freq_sets = prune_candidates(freq_sets, freq_sets, k-1) k += 1 rules = [] for freq_set in freq_sets: for item in freq_set: antecedent = frozenset([item]) consequent = freq_set.difference(antecedent) if len(consequent) > 0: confidence = item_counts[freq_set] / item_counts[antecedent] if confidence >= min_confidence: rules.append((antecedent, consequent, confidence)) return rules # 示例数据 transactions = [ ['A', 'B', 'C'], ['A', 'B'], ['B', 'C'], ['A', 'C'], ['B', 'D'] ] # 调用apriori函数 rules = apriori(transactions, min_support=0.5, min_confidence=0.5) # 打印结果 for antecedent, consequent, confidence in rules: print(f"{antecedent} => {consequent}: {confidence}") ``` 这个示例代码演示了如何使用Apriori来查找频繁项集和关联规则。您可以根据自己的数据和要求进行调整和修改。希望对您有所帮助!
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值