tensorflow：用自己的CSV数据集套用mnist的DBN模型（python实现）

最新推荐文章于 2021-11-09 10:32:46 发布

ErQiu蛋

最新推荐文章于 2021-11-09 10:32:46 发布

阅读量1.3k

点赞数 2

本文链接：https://blog.csdn.net/waterydd/article/details/72867891

版权

本文介绍如何使用深度信念网络（DBN）模型处理流量特征数据，参考MNIST的DBN模型代码，将SiteB.UDP.arff.gz数据集转化为CSV并进行预处理，调整模型参数以适应9个特征和4个类别的分类任务。在尝试运行模型时遇到cost为nan的问题，需要进一步调试和优化。

摘要由CSDN通过智能技术生成

试图用深度学习的DBN模型，来处理流量特征数据集。

（1、选用公共数据集LiWeiDataset——SiteB.UDP.arff.gz，下载网址：http://www.cl.cam.ac.uk/research/srg/netos/projects/brasil/data/index.html

2、模型参考MNIST的DBN模型实现代码，下载地址：https://github.com/xiaohu2015/DeepLearning_tutorials/blob/master/models/dbn.py）

公共数据集的初步处理：数据集共774141个流，提取700000万作为室友数据集，并将数据集分为训练集（600000）和测试集（100000）；其中训练集分为无监督训练集（550000）和验证训练集（50000）。

DBN模型的初步整理：由于参考的DBN模型的通用性，作为我的尝试性代码不需要所有的代码模块，为了代码目的性更清晰易懂，我通过尝试性地删减代码，并保证最终的代码模块仍然能够完整成功的实现 mnist 的分类，最后整理了我所需的代码模块实际上可以就用以下4个 .py

关注