get_one_itemsets.py

get_one_itemsets.py

def get_one_itemsets(transaction_db, min_sup):
    
    items = {}
    
    for t_tup in transaction_db.list_items:
        
        for i in new_t:
            if i in items.keys():
                items[i] += t_tup[1]
            else:
                items[i] = t_tup[1]
    
    # Eliminate one itemsets which do not meet minimum support criteria
    items = {i:sup for (i,sup) in items.items() if sup >= min_sup}
    
    return items