fp-skeleton-partA.py

fp-skeleton-partA.py

def fp_growth(transaction_db, min_sup, fp_list, prefix = []):
    
    ####### STEP A: CONCATENATE #######
    # Get one itemsets which meet minimum support
    one_itemset_dict = get_one_itemsets(transaction_db, min_sup)
    
    # Concanenate frequent patterns with prefix
    freq_patterns = []
    for key,val in one_itemset_dict.items():
        pattern = prefix + [key]
        freq_patterns.append((pattern,val))
    
    for pat in freq_patterns:
        if pat not in fp_list:
            fp_list.append(pat[0])
    
    
    # Step B code to be filled here
  
    # Step C code to be filled here
  

    if cpb_db.shape[0] > 0:
        fp_growth(cpb_db, min_sup, fp_list, prefix + [item])