RuntimeError: Could not infer dtype of numpy.int64

Question

出错代码：
X_for_generate = process.query("target == 1").iloc[:,:-1].values
X_non_default = process.query('target == 0').iloc[:,:-1].values
X_for_generate = torch.tensor(X_for_generate).type(torch.FloatTensor)

其中爆了这行错误：

X_for_generate = torch.tensor(X_for_generate).type(torch.FloatTensor)
RuntimeError: Could not infer dtype of numpy.int64

Answer 1

Gpt辅助回答，我来为你把关

X_for_generate = process.query("
target == 1"
).iloc[:,:-1].values
X_non_default = process.query('target == 0').iloc[:,:-1].values
X_for_generate = torch.tensor(X_for_generate).type(torch.FloatTensor)

这个错误可能是由于torch.tensor无法推断出X_for_generate的数据类型导致的。解决方法可以指定数据类型，例如：

X_for_generate = torch.tensor(X_for_generate, dtype=torch.float32)

这样就能明确指定数据类型，避免出现RuntimeError: Could not infer dtype of numpy.int64的错误。

有问题你别着急，评论留言都可以，看到马上就回复，尽量及时补充齐