pytorch - 无法转换 numpy.bool_ 类型的 np.ndarray

Question

我想知道为什么我在使用时会收到此错误：

Y_train_class = torch.tensor(Y_train_class.values)

TypeError：无法转换 numpy.bool_ 类型的 np.ndarray。唯一支持的类型是：double、float、float16、int64、int32 和 uint8。

我试图将我的数据转换为浮点数，但似乎失败了

X_train = pd.read_csv('c:/Data/x_train_set_yu.csv', header= None)
Y_train = pd.read_csv('c:/Data/y_train_set_yu.1.csv', header= None)

Y_train_class = (Y_train >= 550)
print (Y_train_class)
X_test = pd.read_csv('c:/Data/X_test.csv',header= None)

X_train = torch.tensor(X_train.values)

Y_train.astype(np.float32)
Y_train_class.astype(np.float32)

Y_train_class = torch.tensor(Y_train_class.values)

TypeError：无法转换 numpy.bool_ 类型的 np.ndarray。唯一支持的类型是：double、float、float16、int64、int32 和 uint8。

标签： pytorch

Answer 1

当我Y_train有一个字符串值而不是整数时，我得到了这种类型的错误。用整数替换字符串后，我的错误得到解决。