重庆分公司,新征程启航

为企业提供网站建设、域名注册、服务器等服务

如何用Mxnet和Tensorflow做简单多分类器的对比测试

这篇文章将为大家详细讲解有关如何用Mxnet和Tensorflow做简单多分类器的对比测试,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。

在团风等地区,都构建了全面的区域性战略布局,加强发展的系统性、市场前瞻性、产品创新能力,以专注、极致的服务理念,为客户提供成都网站建设、网站设计 网站设计制作专业公司,公司网站建设,企业网站建设,品牌网站设计,营销型网站建设,外贸网站制作,团风网站建设费用合理。

几个月前,刚开始接触mxnet和Tensorflow时,编写的自己案例使用的就是简单的数据集iris。因为它足够简单(只有3个分类,150条数据),很容易分析。 和很多研究者不太一样,他们只是跑跑训练程序,看看打印出来的train accuracy就结束了。我训练完模型后,还喜欢继续编一些自定义的数据,然后让模型去做预测或分类,看看模型是否足够好。对Iris数据集,我用mxnet实现了一个MLP模型进行分类。在训练完后(打印出来的train accuracy在0.9以上的),马上做一些自定义数据的预测,结果发现预测分类结果很差,没法用。当时觉得是自己不熟悉Mxnet ,所以才会有这样的结果,就不管了。可是,后来转为用Tensorflow实现类似的MLP模型对iris数据集进行分类,然后预测同样的自定义数据,突然发现Tensorflow的代码表现很不错。这种强烈对比差距让我很惊讶,于是又用mxnet和Tensorflow编写MLP模型测试另外两个数据集,结果还是如此(Tensorflow基本可以分类,mxnet不行)。我把一些测试的数据列出来:

测试1. sonar dataset(两个标签值,60个特征,208条数据。):

a)Tensorflow codes :

train Accuracy: 0.833333

测试4个新的数据:

New samples accuracy: 0. 50000

b) mxnet codes:

train Accuracy: 0.6

测试4个新的数据:

('New samples accuracy:', 0.25)

测试2. iris dataset(三个标签值,4个特征,150条数据):

a)Tensorflow codes :

train Accuracy: 1.0

测试10个新的数据:

New samples accuracy: 0.900000

b) mxnet codes:

train Accuracy: 0.975

测试10个新的数据:

('New samples accuracy:', 0.20)

测试3. 自定义数据集(三个标签值,3个特征,10000条数据。这个数据集是我编的,其中前两个特征是无关紧要的,第三列特征才是决定因素。故意这样设置,让神经网络去自己判断谁最重要。

a)Tensorflow codes :

train Accuracy: 0.9

测试10个新的数据:

New samples accuracy: 0.700000.

b) mxnet codes:

train Accuracy: 1.0

测试10个新的数据:

New samples accuracy: 0.3.

可以看到,用mxnet编写的MLP模型对3个不同文本数据集进行新数据分类的精确度很差,没法用。Tensorflow编写的MLP模型能用。mxnet为啥表现不佳呢?我怀疑mxnet代码有bug,特别是模型定义和模型训练这块的代码。在mxnet的github上提了这个问题,但是没人理睬。算了。没找出bug前还是用Tensorflow做这些简单分类的分类器吧。

相关的测试对比代码放在这里:

https://github.com/qinhui99/mxnetSamples.git

关于如何用Mxnet和Tensorflow做简单多分类器的对比测试就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。


当前文章:如何用Mxnet和Tensorflow做简单多分类器的对比测试
URL链接:http://cqcxhl.com/article/pshpep.html

其他资讯

在线咨询
服务热线
服务热线:028-86922220
TOP