Malware Traffic Classification Using Convolutional Neural Network for Representation Learning代码复现的问题

程序谭

已于 2022-05-02 19:50:55 修改

阅读量1.6k

点赞数 4

文章标签： tensorflow 人工智能 python

于 2022-05-02 15:05:33 首次发布

本文链接：https://blog.youkuaiyun.com/chunchenbeile/article/details/124540906

版权

该博客介绍了在尝试复现基于TensorFlow的深度学习代码时遇到的问题及其解决方案，包括输入数据模块导入错误、缺少参数、类转换错误以及InteractiveSession函数的替换等。同时，提供了数据预处理的注意事项和解决办法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

论文下载：https://ieeexplore.ieee.org/document/7899588

代码下载：https://github.com/echowei/DeepTraffic

数据处理工具包使用方法：https://github.com/yungshenglu/USTC-TK2016

作者博士毕业论文收录于知网：基于深度学习的网络流量分类及异常检测方法研究
————————————————
原文链接：https://blog.youkuaiyun.com/caiguanhong/article/details/109695999

在进行代码复现前首先要对数据集进行预处理，预处理部分在其他文章有很多详细介绍，在此不做赘述，本文只简单介绍在复现代码traffic_cnn.py时自己遇到的一些问题。

作者的代码是在tensorflow1.x下写的，我的版本时tf2.9，所以会出现的问题较多。

1.input_data不存在

将import input_data换成import tensorflow.examples.tutorials.mnist.input_data ，若没有tenfowflow.examples，则先在pycharm中新建一个文件

import tensorflow as tf

print(tf.__file__)

查看你的路径。

我的路径是：D:\Anaconda3\envs\TF2.4.0\lib\site-packages\tensorflow，在路径中创建examples文件夹，并将下载好的tutorials包复制进去即可。（tutorials包在网上有很多资源，自行百度下载）

2.报错DATA_DIR = sys.argv[1] IndexError: list index out of range，

这表明没有传入参数，在传入参数DATA_DIR、CLASS_NUM、TRAIN_ROUND时，可以直接写成

DATA_DIR="D:\\code\\tensorflow\\deeptrafic\\DeepTrafficmaster\\1.malware_traffic_classification\\4.TrainAndTest\\5_Mnist"
CLASS_NUM = 10
TRAIN_ROUND = 400

这种形式，也可以在pycharm中自己设置，点击“Run”菜单下的"Edit Configurations..."命令，在parameter中输入三个参数的值，每个参数输入完后，需要有一个空格。