【发布时间】:2020-06-19 12:36:10
【问题描述】:
我目前正在使用多层感知器 (MLP) 开发一种有监督的异常检测,目标是在良性流量和恶意流量之间进行分类。我用的是CTU-13 dataset,数据集的样本如下: Sample of Dataset。该数据集有 169032 个良性流量和 143828 个恶意流量。我的 MLP 模型的代码如下:
def MLP_model():
model = Sequential()
model.add(Dense(1024,input_dim=15, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(256,activation='relu'))
model.add(Dense(256,activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(128,activation='relu'))
model.add(Dense(128,activation='relu'))
model.add(Dense(1, activation='sigmoid'))
adam = optimizers.Adam(lr=0.0001, beta_1=0.9, beta_2=0.999, epsilon=None, decay=0.0, amsgrad=False)
model.compile(optimizer = adam, loss='binary_crossentropy', metrics=['accuracy'])
return model
model = MLP_model()
#With Callbacks
callbacks = [EarlyStopping('val_loss', patience=5)]
hist = model.fit(Xtrain, Ytrain, epochs=50, batch_size=50, validation_split=0.20, callbacks=callbacks, verbose=1)
我得到的结果如下:
Accuracy: 0.923045
Precision: 0.999158
Recall: 0.833308
F1 score: 0.908728
但是,从训练曲线来看,我怀疑模型拟合不足(基于this article): The Model's Training Curve
我尝试增加神经元和层数(建议here),但同样的问题仍然存在。感谢您为解决此问题提供的任何帮助。
【问题讨论】:
-
你能分享详细信息吗?您的问题似乎更加过拟合,因为您的验证损失远高于火车损失。所以我实际上会将神经元的数量减少 4 倍。另外,你是否在 0 和 1 之间缩放你的输入?
标签: python tensorflow keras neural-network mlp