根据我的测试,Keras中的to_categorical()返回ndarray of float64。我想知道为什么在默认情况下它不是float32,它可以由GPU处理。of to_categorical()没有说明返回类型应该是什么。因此,我想这可能是一个实现细节,而不是协议/接口的一部分。总之,有两个问题:
不应该依赖返回类型,因为它可能不是接口的一部分,因此可能会发生更改吗?如果是这样的话,如果关心类型,则应该始终追加astype()调用
我的列车数据如下所示:categorial=[c for c in train.columns if train.columns(c).dtype in ['object'] ]---------------------------------------------------------------------------<ipython-input-31-3