Poda básica de resultados de Apriori

Análisis de cesta de la compra en Python

Isaiah Hull

Visiting Associate Professor of Finance, BI Norwegian Business School

Apriori y reglas de asociación

Apriori poda itemsets.
- Aplica un umbral mínimo de soporte.
- Una versión modificada puede podar por número de ítems.
- No nos dice nada sobre reglas de asociación.
Reglas de asociación.
- Hay muchas más reglas que itemsets.
- {Bolsas, Cajas}: Bolsas -> Cajas O Cajas -> Bolsas.

Cómo calcular reglas de asociación

Calcular reglas desde Apriori.
- Difícil de enumerar para n y k altos.
- Podría deshacer la poda de itemsets de Apriori.

Reducir el número de reglas.
- El módulo mlxtend permite podar reglas de asociación.
- association_rules() recibe itemsets frecuentes, métrica y umbral.

Cómo calcular reglas de asociación

# Import Apriori algorithm
from mlxtend.frequent_patterns import apriori, association_rules

# Load one-hot encoded novelty gifts data
onehot = pd.read_csv('datasets/online_retail_onehot.csv')

# Apply Apriori algorithm
frequent_itemsets = apriori(onehot, 
                            use_colnames=True, 
                            min_support=0.0001)

# Compute association rules
rules = association_rules(frequent_itemsets,
                          metric = "support", 
                          min_threshold = 0.0)

La importancia de podar

# Print the rules.
print(rules)

                               antecedents  ... conviction
0      (CARDHOLDER GINGHAM CHRISTMAS TREE)  ...      inf
...
79505      (SET OF 3 HEART COOKIE CUTTERS)  ... 1.998496

# Print the frequent itemsets.
print(frequent_itemsets)

       support                                           itemsets
0     0.000752                   ( 50'S CHRISTMAS GIFT BAG LARGE)
...
4707  0.000752                  (PIZZA PLATE IN BOX, CHRISTMAS ...

La importancia de podar

# Compute association rules
rules = association_rules(frequent_itemsets,
                          metric = "support", 
                          min_threshold = 0.001)

# Print the rules.
print(rules)

                   antecedents      conviction  
0  (BIRTHDAY CARD, RETRO SPOT)  ...  2.977444 
1    (JUMBO BAG RED RETROSPOT)  ...  1.247180

Explorar el conjunto de reglas

print(rules.columns)

Index(['antecedents', 'consequents', 'antecedent support',
       'consequent support', 'support', 'confidence', 'lift', 'leverage',
       'conviction'],
      dtype='object')

print(rules[['antecedents','consequents']])

                   antecedents                  consequents
0    (JUMBO BAG RED RETROSPOT)  (BIRTHDAY CARD, RETRO SPOT)
1  (BIRTHDAY CARD, RETRO SPOT)    (JUMBO BAG RED RETROSPOT)

Poda con otras métricas

# Compute association rules
rules = association_rules(frequent_itemsets,
                          metric = "antecedent support", 
                          min_threshold = 0.002)

# Print the number of rules.
print(len(rules))

¡Vamos a practicar!

Análisis de cesta de la compra en Python