利用CNN代码复现论文
流程
第一,对原始数据进行归一化处理等预处理
- 第一步类型转化,数据集中,有三个字符特征,是
protocol_type
、service
和flag
,由于他们是字符特征所以需要进行one-hot
编码,从而把字符型转为数值型 - 第二步对数据做归一化处理,把所有数据的特征缩放到
0
和1
之间,使用的处理方法为min-max
方法
第二,图片转化
进行输入准备,因为CNN
的输入是图片形式,先把归一化之后的122
个特征变换成矩阵的形式,大小是一个十二乘十二的方阵,然后进行像素化处理,将矩阵中的数据进行与255
相乘的操作,形成一张图片。
第三,构建模型进行分类计算
详细实现
PreHandle
对每行数据遍历,将四个关键字分别创建匹配列表,四个关键字替换为该关键字在匹配列表种的索引值