TensorFlow估计器的类别数量不变

我尝试使用TensorFlow估计器来处理MNIST数据集。不知为何,它一直显示我的n_classes被设置为1,尽管实际上我设置的是10!

import tensorflow as tfimport numpy as npfrom tensorflow.examples.tutorials.mnist import input_datamnist = input_data.read_data_sets("MNIST_data/",one_hot=True)feature_columns = [tf.feature_column.numeric_column("x", shape=[784])]# 构建一个具有10、20、10个单元的3层DNN。classifier = tf.estimator.DNNClassifier(feature_columns=feature_columns,                                        hidden_units=[500, 500, 500],                                        n_classes=10,                                        model_dir="/tmp/MT")for i in range(100000):    xdata, ydata = mnist.train.next_batch(500)    train_input_fn = tf.estimator.inputs.numpy_input_fn(        x={"x":xdata},        y=ydata,        num_epochs=None,        shuffle=True)    classifier.train(input_fn=train_input_fn, steps=2000)# 定义测试输入test_input_fn = tf.estimator.inputs.numpy_input_fn(    x= {"x":mnist.test.images},    y= mnist.test.labels,    num_epochs=1,    shuffle=False)# 评估准确率accuracy_score = classifier.evaluate(input_fn=test_input_fn)["accuracy"]print("\n测试准确率: {0:f}\n".format(accuracy_score))

错误:

ValueError: 标签形状不匹配。分类器配置为n_classes=1。收到的却是10。建议修复:检查您传递给估计器的n_classes参数和/或标签的形状。程序以退出码1结束

回答:

这是一个好问题。tf.estimator.DNNClassifier使用的是tf.losses.sparse_softmax_cross_entropy损失,换句话说,它期望使用序数编码,而不是独热编码(文档中没有找到相关说明,只能在源代码中找到):

labels必须是一个密集的Tensor,其形状与logits匹配,即[D0, D1, ... DN, 1]。如果给定了label_vocabularylabels必须是一个字符串Tensor,其值来自词汇表。如果未给出label_vocabularylabels必须是一个整数Tensor,其值指定类别索引。

您应该使用one_hot=False读取数据,并且将标签转换为int32类型以使其正常工作:

y=ydata.astype(np.int32)

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注