1

skflow で非常に単純な nn 分類器を実行しようとしています。

classifier = skflow.TensorFlowDNNClassifier(
hidden_units=[10, 10, 10],
n_classes=10,
batch_size=100,
learning_rate=0.05)
print (data.train.images).shape
print (data.train.labels).shape
classifier.fit(data.train.images,data.train.labels)

出力は: (73257, 3072) (73257, 10)

エラーは次のとおりです。

in assert_same_rank
    "Shapes %s and %s must have the same rank" % (self, other))
ValueError: Shapes (?, 10) and (?, 10, 10) must have the same rank

ここで何が問題なのかよくわかりません:(

4

1 に答える 1