DGA系列之RNN(一)

这篇博客是对书中《web安全深度学习实战》DGA域名检测RNN的代码进行复现与解释。

实验步骤

  1. 获取样本数据
  2. 提取特征
  3. 将样本划分为训练集和测试集
  4. 使用RNN算法在训练集上训练,获得模型数据
  5. 使用模型数据在测试集上进行测试
  6. 验证RNN算法的结果

获取样本数据

在这里插入图片描述
老规矩,加载正常样本和黑样本。

提取特征与划分数据集

在这里插入图片描述

训练、预测和验证RNN

在这里插入图片描述
将提取出来的特征数据二值化,并定义长度为64,不够长度就补0

在这里插入图片描述
建立网络模型,先定义传入数据形状,64维,然后通过embedding再嵌入,变成64x64,并且也将张量传入lstm。最后连接全层网络,定义损失函数等。
在这里插入图片描述
训练数据

在这里插入图片描述
验证数据,最后输出报告
在这里插入图片描述
吐槽一句,这里我用我的8核心cpu跑这个代码,直接爆炸…
后来感谢有个大佬借了台32核的服务器跑了一个钟才跑完>-<,结果不易啊

代码上传到个人github:rnn_dga.py

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值