使用 Python 进行关联规则挖掘和神经网络分类

使用 Python 进行关联规则挖掘和神经网络分类

import pandas as pd
from mlxtend.preprocessing import TransactionEncoder
from mlxtend.frequent_patterns import apriori, association_rules
from sklearn.neural_network import MLPClassifier

# 读取数据集
df = pd.read_csv('milk_bread.csv')

# 将数据集转换为事务列表
te = TransactionEncoder()
te_ary = te.fit_transform(df.values)
transactions = pd.DataFrame(te_ary, columns=te.columns_)

# 训练神经网络
clf = MLPClassifier(hidden_layer_sizes=(10,), max_iter=1000, random_state=0)
clf.fit(transactions, transactions)

# 提取关联规则
frequent_itemsets = apriori(transactions, min_support=0.1, use_colnames=True)
rules = association_rules(frequent_itemsets, metric='lift', min_threshold=1)

# 打印关联规则
print(rules)

实现的功能

该代码实现了以下功能：

1. 读取名为 'milk_bread.csv' 的数据集。
2. 将数据集转换为事务列表。
3. 使用 MLxtend 库中的 apriori 算法寻找频繁项集。
4. 使用 MLxtend 库中的 association_rules 函数根据指定的度量标准 (lift) 和最小阈值 (1) 提取关联规则。
5. 使用 sklearn 库中的 MLPClassifier 训练一个神经网络模型。
6. 打印提取出的关联规则。