python-2.7 - Theano TensorType 错误

Question

当我使用nolearn实现多标签分类时，出现了这个错误:

'Bad input argument to theano function with name "/Users/lm/Documents/anaconda/lib/python2.7/site-packages/nolearn/lasagne/base.p‌​y:391" at index 1(0-based)', 'TensorType(float32, matrix) cannot store a value of dtype int64 without risking loss of precision. If you do not mind this loss, you can: 1) explicitly cast your data to float32, or 2) set "allow_input_downcast=True" when calling "function".', array([[0, 0, 0, ..., 0, 0, 1],

Answer 1

如错误消息中所述，您需要将输入和输出转换为适当的类型(如果您不担心失去精度)。

input = input.astype(np.float32)
output = output.astype(np.float32)

应该可以

注意:即使您这样做，如果您有一个转换数据的 BatchIterator(并且不小心再次使用了 float64)，错误可能仍然存在。解决方案是相同的:在 BatchIterator 中，在返回数据之前立即将数据转换为 float32。