获取数组输出但我想得到一个使用稀疏分类损失的输出

我正在尝试用Keras训练一个简单的neural net。我有输入数据,并且希望得到一个整数输出,这个整数代表一个类别。我希望这个整数在0到13的范围内。然而,当最后一个输出设置为1时,我会收到一个错误

InvalidArgumentError: Received a label value of 12 which is outside the valid range of [0, 1).  Label values:

这是我目前编译神经网络的代码

我的训练输入是数组的数组,标签只是一个包含0到12的值的数组

这是输出

enter image description here


回答:

让我们理解稀疏分类交叉熵

它只会让你通过整数标签(而不是one-hot数组)来测量错误。

那么为什么会出错?

根据我所解释的,你的网络应该能够预测14个类别。因此,像one-hot编码所做的事情一样,网络也需要这样做(并不是说你正在输入one-hot,这只是一个回顾那个方法,以提醒我们应该做什么),你需要14个输出神经元来实现这一点,因此;最后一层应该看起来像这样:

classifier.add(Dense(units = 14, kernel_initializer = 'uniform', activation = 'sigmoid'))

顺便说一句,使用metrics = ['sparse_categorical_accuracy']是一个好习惯

如果你想要整数输出

据我所知,有两个选项:

y_pred = np.argmax(classifier.predict(X_test), axis=1)

或者简单地:

y_pred = classifier.predict_classes(X_test)

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中创建了一个多类分类项目。该项目可以对…

发表回复

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