TFLearn не может правильно загрузить данные тренировки с формой

Question

Я пытаюсь создать ИИ, чтобы предсказать исход матчей FRC с использованием тензорного потока и TFLearn.

Вот соответствующая

x = np.load("FRCPrediction/matchData.npz")["x"]
y = np.load("FRCPrediction/matchData.npz")["y"]

def buildModel():
    net = tflearn.input_data([10, 0])
    net = tflearn.fully_connected(net, 64)
    net = tflearn.dropout(net, 0.5)
    net = tflearn.fully_connected(net, 10, activation='softmax')
    net = tflearn.regression(net, optimizer='adam', loss='categorical_crossentropy')

    model = tflearn.DNN(net)
    return model

model = buildModel()

BATCHSIZE = 128

model.fit(x, y, batch_size = BATCHSIZE)

Ошибка с ошибкой:

Training samples: 36024
Validation samples: 0
--
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-12-ce7cbb8e618a> in <module>()
----> 1 model.fit(x, y, batch_size = BATCHSIZE)

4 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/client/session.py in _run(self, handle, fetches, feed_dict, options, run_metadata)
   1126                              'which has shape %r' %
   1127                              (np_val.shape, subfeed_t.name,
-> 1128                               str(subfeed_t.get_shape())))
   1129           if not self.graph.is_feedable(subfeed_t):
   1130             raise ValueError('Tensor %s may not be fed.' % subfeed_t)

ValueError: Cannot feed value of shape (128, 36) for Tensor 'InputData/X:0', which has shape '(?, 10, 0)

Любая помощь очень ценится.Спасибо.

Answer 1

Эта ошибка означает, что ваш размер X равен (some_length, 36), что не может соответствовать входному слою с размером (10, 0). Я сомневаюсь, что для вашего второго измерения, равного 0, форма должна быть не менее 1. Для ее решения необходимо сделать:

net = tflearn.input_data(shape=[None, 36])

None для динамического измерения, которое будет соответствовать всем BATCHSIZE, независимо от того, 128, 1000 или 2000