最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

TensorFlow

运维笔记admin42浏览0评论

TensorFlow

TensorFlow

小智 5

因为我看不到你的 trainY 编码;看起来 - 你的 trainY 只有一列,你的模型输出有 10 个神经元,所以 Shapes (None, 1) 和 (None, 10) 是不兼容的。你可以在你的 trainY 上试试这个(即 one-hot 编码)

from sklearn.preprocessing import LabelBinarizerlabel_as_binary = LabelBinarizer()train__y_labels = label_as_binary.fit_transform(trainY)

编译看起来像(寻找train__y_labels)

model.pile(optimizer='adam', loss='categorical_crossentropy', metrics=['auracy'])model.fit(train_X_input, train__y_labels, batch_size=128, epochs=1)

注意:如果您的 valid 也引发错误,则所有 y(s) 都需要相同。

TensorFlow

发布评论

评论列表(0)

  1. 暂无评论