第十二届“中国软件杯”大赛：A10-基于机器学习的分布式系统故障诊断系统——baseline（二）

于 2023-04-27 15:08:30 发布
阅读量3.2k 收藏 44
点赞数 9
分类专栏： 机器学习 文章标签： 机器学习 人工智能 数据挖掘 分类
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jm863936705/article/details/130407040
版权

赛题介绍

在分布式系统中某个节点发生故障时，故障会沿着分布式系统的拓扑结构进行传播，造成自身节点及其邻接节点相关的KPI指标和发生大量日志异常。本次比赛提供分布式数据库的故障特征数据和标签数据，其中特征数据是系统发生故障时的KPI指标数据，KPI指标包括由feature0、feature1 …feature106共107个指标，标签数据为故障类别数据，共6个类别，用0、1、2、3、4、5分别表示6个故障，参赛人员可根据这些数据，借助机器学习、深度学习、web等技术搭建故障诊断系统，该系统支持用户上传训练集对模型进行训练和模型下载，同时支持用户上传单条或多条测试语句进行测试并可视化测试结果，支持测试结果下载。

baseline: RandomForest

数据分析

读取数据

df = pd.read_csv('data/train/train.csv', index_col=None)

判断是否有缺失值

df.isnull().any()
'''
output: True即为存在缺失值
sample_id     False
feature0       True
feature1       True
feature2       True
feature3       True
              ...  
feature103     True
feature104     True
feature105    False
feature106     True
label         False
Length: 109, dtype: bool
'''

数据标准化及缺失值填充

# 数据标准化
features = df.iloc[:, 1:-1]
numeric_features = features.dtypes[features.dtypes != 'object'].index
features[numeric_features] = features[numeric_features].apply(
    lambda x: (x - x.mean()) / (x.std())
)
# 在标准化数据之后，所有均值消失，因此我们可以将缺失值设置为0
features[numeric_features] = features[numeric_features].fillna(0)
features_labels = pd.concat([features, df[['label']]], axis=1)
train_features = pd.concat([df[['sample_id']], features], axis=1)
train_label = df[['sample_id', 'label']]
df = pd.concat([train_features, train_label[['label']]], axis=1)

观察数据基本信息

# 观察前五行数据
df.head()

在这里插入图片描述

# 数据大小
df.shape
'''
output:
(6296, 109)
'''

df.dtypes
'''
output:
sample_id       int64
feature0      float64
feature1      float64
feature2      float64
feature3      float64
               ...   
feature103    float64
feature104    float64
feature105    float64
feature106    float64
label           int64
Length: 109, dtype: object
'''

# 类别分布
df['label'].value_counts().sort_index().plot(kind='bar')
plt.show()

df['label'].value_counts().sort_index().plot(kind='pie')
plt.show()

在这里插入图片描述
在这里插入图片描述

features.describe()

在这里插入图片描述

# 分组的平均数据统计
label_Summary = features_labels.groupby('label')
label_Summary.mean()

在这里插入图片描述

# 相关性矩阵
corr = features_labels.corr()
sns.set_context({'figure.figsize':[100, 100]})
fig = sns.heatmap(corr, 
                xticklabels=corr.columns.values, 
                yticklabels=corr.columns.values)
heatmap = fig.get_figure()
heatmap.savefig('work/heatmap.png', dpi=300)
corr

在这里插入图片描述
在这里插入图片描述

# 各个特征的概率密度函数
feature_names = features.columns.values.tolist()
for name in feature_names:
    fig = plt.figure(figsize=(15, 4), )
    ax = sns.kdeplot(df.loc[(df['label'] == 0), name], color='b', shade=True, label='0')
    ax = sns.kdeplot(df.loc[(df['label'] == 1), name], color='r', shade=True, label='1')
    ax = sns.kdeplot(df.loc[(df['label'] == 2), name], color='g', shade=True, label='2')
    ax = sns.kdeplot(df.loc[(df['label'] == 3), name], color='y', shade=True, label='3')
    ax = sns.kdeplot(df.loc[(df['label'] == 4), name], color='m', shade=True, label='4')
    ax = sns.kdeplot(df.loc[(df['label'] == 5), name], color='c', shade=True, label='5')
    ax.set(xlabel=name, ylabel='频率')
    plt.title('{} Probabilitydensity function'.format(name))
    plt.savefig('work/{}的概率密度函数图.png'.format(name))

在这里插入图片描述

划分数据集

from sklearn.model_selection import train_test_split

target_name = 'label'
x = df.drop(['sample_id', 'label'], axis=1)
y = df[['label']]

x_train, x_test, y_train, y_test = train_test_split(
    x, y, test_size=0.15, random_state=123, stratify=y)

评价指标

# 指标计算 参数：array
def metrics_calculate(pred, y_test, txt_path):
    TP = [0, 0, 0, 0, 0, 0]
    FP = [0, 0, 0, 0, 0, 0]
    FN = [0, 0, 0, 0, 0, 0]
    for i in range(len(y_test)):
        if pred[i] == 0 and y_test[i] == 0:
            TP[0] += 1
        if pred[i] != 0 and y_test[i] == 0:
            FN[0] += 1
        if pred[i] == 0 and y_test[i] != 0:
            FP[0] += 1

        if pred[i] == 1 and y_test[i] == 1:
            TP[1] += 1
        if pred[i] != 1 and y_test[i] == 1:
            FN[1] += 1
        if pred[i] == 1 and y_test[i] != 1:
            FP[1] += 1

        if pred[i] == 2 and y_test[i] == 2:
            TP[2] += 1
        if pred[i] != 2 and y_test[i] == 2:
            FN[2] += 1
        if pred[i] == 2 and y_test[i] != 2:
            FP[2] += 1

        if pred[i] == 3 and y_test[i] == 3:
            TP[3] += 1
        if pred[i] != 3 and y_test[i] == 3:
            FN[3] += 1
        if pred[i] == 3 and y_test[i] != 3:
            FP[3] += 1

        if pred[i] == 4 and y_test[i] == 4:
            TP[4] += 1
        if pred[i] != 4 and y_test[i] == 4:
            FN[4] += 1
        if pred[i] == 4 and y_test[i] != 4:
            FP[4] += 1

        if pred[i] == 5 and y_test[i] == 5:
            TP[5] += 1
        if pred[i] != 5 and y_test[i] == 5:
            FN[5] += 1
        if pred[i] == 5 and y_test[i] != 5:
            FP[5] += 1

    Precision = [0, 0, 0, 0, 0, 0]
    Recall = [0, 0, 0, 0, 0, 0]
    F1 = [0, 0, 0, 0, 0, 0]

    if (TP[0] + FP[0]) == 0:
        Precision[0] = 0.0
    else:
        Precision[0] = TP[0] / (TP[0] + FP[0])
    if (TP[1] + FP[1]) == 0:
        Precision[1] = 0.0
    else:
        Precision[1] = TP[1] / (TP[1] + FP[1])
    if (TP[2] + FP[2]) == 0:
        Precision[2] = 0.0
    else:
        Precision[2] = TP[2] / (TP[2] + FP[2])
    if (TP[3] + FP[3]) == 0:
        Precision[3] = 0.0
    else:
        Precision[3] = TP[3] / (TP[3] + FP[3])
    if (TP[4] + FP[4]) == 0:
        Precision[4] = 0.0
    else:
        Precision[4] = TP[4] / (TP[4] + FP[4])
    if (TP[5] + FP[5]) == 0:
        Precision[5] = 0.0
    else:
        Precision[5] = TP[5] / (TP[5] + FP[5])

    # for i in range(6):
    #     print('Precision: {}\n'.format(Precision[i]))

    if (TP[0] + FN[0]) == 0:
        Recall[0] = 0.0
    else:
        Recall[0] = TP[0] / (TP[0] + FN[0])
    if (TP[1] + FN[1]) == 0:
        Recall[1] = 0.0
    else:
        Recall[1] = TP[1] / (TP[1] + FN[1])
    if (TP[2] + FN[2]) == 0:
        Recall[2] = 0.0
    else:
        Recall[2] = TP[2] / (TP[2] + FN[2])
    if (TP[3] + FN[3]) == 0:
        Recall[3] = 0.0
    else:
        Recall[3] = TP[3] / (TP[3] + FN[3])
    if (TP[4] + FN[4]) == 0:
        Recall[4] = 0.0
    else:
        Recall[4] = TP[4] / (TP[4] + FN[4])
    if (TP[5] + FN[5]) == 0:
        Recall[5] = 0.0
    else:
        Recall[5] = TP[5] / (TP[5] + FN[5])

    # for i in range(6):
    #     print('Recall: {}\n'.format(Recall[i]))

    if (Precision[0] + Recall[0]) == 0:
        F1[0] = 0.0
    else:
        F1[0] = (2 * Precision[0] * Recall[0]) / (Precision[0] + Recall[0])
    if (Precision[1] + Recall[1]) == 0:
        F1[1] = 0.0
    else:
        F1[1] = (2 * Precision[1] * Recall[1]) / (Precision[1] + Recall[1])
    if (Precision[2] + Recall[2]) == 0:
        F1[2] = 0.0
    else:
        F1[2] = (2 * Precision[2] * Recall[2]) / (Precision[2] + Recall[2])
    if (Precision[3] + Recall[3]) == 0:
        F1[3] = 0.0
    else:
        F1[3] = (2 * Precision[3] * Recall[3]) / (Precision[3] + Recall[3])
    if (Precision[4] + Recall[4]) == 0:
        F1[4] = 0.0
    else:
        F1[4] = (2 * Precision[4] * Recall[4]) / (Precision[4] + Recall[4])
    if (Precision[5] + Recall[5]) == 0:
        F1[5] = 0.0
    else:
        F1[5] = (2 * Precision[5] * Recall[5]) / (Precision[5] + Recall[5])

    # for i in range(6):
    #     print('F1: {}\n'.format(F1[i]))

    Macro_Precision = sum([Precision[0], Precision[1], Precision[2],
                            Precision[3], Precision[4], Precision[5]]) / 6
    Macro_Recall = sum([Recall[0], Recall[1], Recall[2],
                        Recall[3], Recall[4], Recall[5]]) / 6
    Macro_F1 = sum([F1[0], F1[1], F1[2], F1[3], F1[4], F1[5]]) / 6

    l_sum = sum([TP[0], TP[1], TP[2], TP[3], TP[4], TP[5]])
    m_sum = l_sum + sum([FP[0], FP[1], FP[2], FP[3], FP[4], FP[5]])
    n_sum = l_sum + sum([FN[0], FN[1], FN[2], FN[3], FN[4], FN[5]])

    if m_sum == 0:
        Micro_Precision = 0.0
    else:
        Micro_Precision = l_sum / m_sum
    # print('Micro_Precision: {}\n'.format(Micro_Precision))
    if n_sum == 0:
        Micro_Recall = 0.0
    else:
        Micro_Recall = l_sum / n_sum
    # print('Micro_Recall: {}\n'.format(Micro_Recall))
    if (Micro_Precision + Micro_Recall) == 0:
        Micro_F1 = 0.0
    else:
        Micro_F1 = (2 * Micro_Precision * Micro_Recall) / (Micro_Precision + Micro_Recall)
    # print('Micro_F1: {}\n'.format(Micro_F1))

    if txt_path == None:
        return '%.4f' % Micro_F1
    else:
        f = open(txt_path, 'a', encoding='utf-8')
        for i in range(6):
            f.write('类别{}: '.format(i))
            f.write('\n')
            f.write('Precision: {:.2f}%'.format(Precision[i] * 100))
            f.write('\n')
            f.write('Recall: {:.2f}%'.format(Recall[i] * 100))
            f.write('\n')
            f.write('F1: {:.2f}'.format(F1[i]))
            f.write('\n')
        f.write('Macro_Precision: {:.2f}%'.format(Macro_Precision * 100))
        f.write('\n')
        f.write('Macro_Recall: {:.2f}%'.format(Macro_Recall * 100))
        f.write('\n')
        f.write('Macro_F1: {:.2f}'.format(Macro_F1))
        f.write('\n')
        f.write('Micro_Precision: {:.2f}%'.format(Micro_Precision * 100))
        f.write('\n')
        f.write('Micro_Recall: {:.2f}%'.format(Micro_Recall * 100))
        f.write('\n')
        f.write('Micro_F1: {:.2f}'.format(Micro_F1))
        f.write('\n')
        f.close()

模型训练

# 实例化随机森林
rf = RandomForestClassifier(
    criterion='entropy',
    n_estimators=3,
    max_depth=None,  # 定义树的深度，可以用来防止过拟合
    min_samples_split=10,  # 定义至少多少个样本的情况下才继续分叉
    min_samples_leaf=0.02  # 定义叶子节点最少需要包含多少个样本（百分比表达），防止过拟合
)
# 模型训练
rf.fit(x_train, y_train)

指标计算

# 先对t_test（DataFrame）进行reshape
y_test = y_test.values.reshape(-1, )
rf_pred = rf.predict(x_test)
txt_path = 'work/result_RandomForest.txt'
metrics_calculate(rf_pred, y_test, txt_path)

'''
output:
Precision: 0.7547169811320755

Precision: 0.7230769230769231

Precision: 0.7387387387387387

Precision: 1.0

Precision: 0.8297872340425532

Precision: 1.0

Recall: 0.9856262833675564

Recall: 0.47474747474747475

Recall: 0.5394736842105263

Recall: 0.3132530120481928

Recall: 0.7358490566037735

Recall: 0.8450704225352113

F1: 0.854853072128228

F1: 0.573170731707317

F1: 0.623574144486692

F1: 0.47706422018348627

F1: 0.78

F1: 0.916030534351145

Micro_Precision: 0.7767195767195767

Micro_Recall: 0.7767195767195767

Micro_F1: 0.7767195767195768
'''

随机森林可视化

# 随机森林可视化
Estimators = rf.estimators_
class_names = ['0', '1', '2', '3', '4', '5']
feature_names = df.columns[1:-1]
for index, model in enumerate(Estimators):
    dot_data = StringIO()
    export_graphviz(model, out_file=dot_data,
                    feature_names=feature_names,
                    class_names=class_names,
                    filled=True, rounded=True, 
                    special_characters=True)
    graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
    graph.write_png('work/Rf{}.png'.format(index))
    plt.figure(figsize=(20, 20))
    plt.imshow(plt.imread('work/Rf{}.png'.format(index)))
    plt.axis('off')

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

哎呦哥哥、
关注
专栏目录
评论 5
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

哎呦哥哥、

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

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

余额充值