Theano TensorType错误

Question

当我使用nolearn实现多标签分类时,我收到此错误:

'在索引1(基于0)的名称为"/Users/lm/Documents/anaconda/lib/python2.7/site-packages/nolearn/lasagne/base.p y:391"的theano函数输入参数错误,'TensorType(float32,matrix)无法存储dtype int64的值而不会有丢失精度的风险.如果你不介意这种损失,你可以:1)显式地将你的数据转换为float32,或者2)在调用"function"时设置"allow_input_downcast = True".',array([[0,0,0,... ,0,0,1],

Answer 1

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

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

应该管用

注意:即使您这样做,如果您有一个BatchIterator转换数据(并且无意中float64再次使用),则错误可能仍然存在.解决方案是相同的:在返回之前BatchIterator,将数据转换为float32右边.