inputs=pd.get_dummies(inputs,dummy_na=True,dtype=int)结果还还是bool型
时间: 2024-09-19 10:08:08 浏览: 194
当你使用pandas的`get_dummies`函数将类别变量转换为虚拟/哑变量(one-hot encoding)时,如果设置了`dummy_na=True`,它会在缺失值的地方添加一个新的虚拟列表示存在缺失。默认情况下,这些新添加的列会被赋予整数类型,通常是0代表缺失,非缺失的值会用1或其他数字标识。
如果你希望所有输出都是布尔类型(True/False),而不是整数,你需要在调用`get_dummies`之后对结果进行进一步处理。例如,你可以直接检查每个列是否等于缺失值的特殊标记(在这种情况下可能是0),然后将它们转换成布尔类型。可以这样做:
```python
import pandas as pd
# 假设df是包含输入数据的DataFrame
dummies = pd.get_dummies(inputs, dummy_na=True, dtype=int)
booleans = (dummies != 0).astype(bool)
# 或者,如果你想要更简洁的方式,可以链式操作:
booleans = (pd.get_dummies(inputs, dummy_na=True, dtype=int) != 0).astype(bool)
```
相关问题
def calc_gradient_penalty(self, netD, real_data, fake_data): alpha = torch.rand(1, 1) alpha = alpha.expand(real_data.size()) alpha = alpha.cuda() interpolates = alpha * real_data + ((1 - alpha) * fake_data) interpolates = interpolates.cuda() interpolates = Variable(interpolates, requires_grad=True) disc_interpolates, s = netD.forward(interpolates) s = torch.autograd.Variable(torch.tensor(0.0), requires_grad=True).cuda() gradients1 = autograd.grad(outputs=disc_interpolates, inputs=interpolates, grad_outputs=torch.ones(disc_interpolates.size()).cuda(), create_graph=True, retain_graph=True, only_inputs=True, allow_unused=True)[0] gradients2 = autograd.grad(outputs=s, inputs=interpolates, grad_outputs=torch.ones(s.size()).cuda(), create_graph=True, retain_graph=True, only_inputs=True, allow_unused=True)[0] if gradients2 is None: return None gradient_penalty = (((gradients1.norm(2, dim=1) - 1) ** 2).mean() * self.LAMBDA) + \ (((gradients2.norm(2, dim=1) - 1) ** 2).mean() * self.LAMBDA) return gradient_penalty def get_loss(self, net,fakeB, realB): self.D_fake, x = net.forward(fakeB.detach()) self.D_fake = self.D_fake.mean() self.D_fake = (self.D_fake + x).mean() # Real self.D_real, x = net.forward(realB) self.D_real = (self.D_real+x).mean() # Combined loss self.loss_D = self.D_fake - self.D_real gradient_penalty = self.calc_gradient_penalty(net, realB.data, fakeB.data) return self.loss_D + gradient_penalty,return self.loss_D + gradient_penalty出现错误:TypeError: unsupported operand type(s) for +: 'Tensor' and 'NoneType'
这个错误是因为在 calc_gradient_penalty 函数中,如果 gradients2 是 None,则会返回 None。在 get_loss 函数中,如果 gradient_penalty 是 None,则返回的就是 self.loss_D + None,这导致了 TypeError 错误。
为了解决这个问题,你可以在 calc_gradient_penalty 函数中添加一个条件判断,如果 gradients2 是 None,则返回 0.0,而不是 None。这样,当计算 gradient_penalty 时,即使 gradients2 是 None,也不会导致错误。修改后的代码如下:
```
def calc_gradient_penalty(self, netD, real_data, fake_data):
alpha = torch.rand(1, 1)
alpha = alpha.expand(real_data.size())
alpha = alpha.cuda()
interpolates = alpha * real_data + ((1 - alpha) * fake_data)
interpolates = interpolates.cuda()
interpolates = Variable(interpolates, requires_grad=True)
disc_interpolates, s = netD.forward(interpolates)
s = torch.autograd.Variable(torch.tensor(0.0), requires_grad=True).cuda()
gradients1 = autograd.grad(outputs=disc_interpolates, inputs=interpolates,
grad_outputs=torch.ones(disc_interpolates.size()).cuda(),
create_graph=True, retain_graph=True, only_inputs=True, allow_unused=True)[0]
gradients2 = autograd.grad(outputs=s, inputs=interpolates,
grad_outputs=torch.ones(s.size()).cuda(),
create_graph=True, retain_graph=True,
only_inputs=True, allow_unused=True)[0]
if gradients2 is None:
return 0.0
gradient_penalty = (((gradients1.norm(2, dim=1) - 1) ** 2).mean() * self.LAMBDA) + \
(((gradients2.norm(2, dim=1) - 1) ** 2).mean() * self.LAMBDA)
return gradient_penalty
def get_loss(self, net,fakeB, realB):
self.D_fake, x = net.forward(fakeB.detach())
self.D_fake = self.D_fake.mean()
self.D_fake = (self.D_fake + x).mean()
# Real
self.D_real, x = net.forward(realB)
self.D_real = (self.D_real+x).mean()
# Combined loss
self.loss_D = self.D_fake - self.D_real
gradient_penalty = self.calc_gradient_penalty(net, realB.data, fakeB.data)
if gradient_penalty == None:
gradient_penalty = 0.0
return self.loss_D + gradient_penalty
```
opt = config.get_arguments().parse_args() netG = Generator(opt) for param in netG.parameters(): param.requires_grad = False ckpt_dir = os.path.join(opt.checkpoints, opt.dataset, opt.attack_mode, 'target_' + str(opt.target_label)) ckpt_path = os.path.join(ckpt_dir, "{}_{}_ckpt.pth.tar".format(opt.attack_mode, opt.dataset)) state_dict = torch.load(ckpt_path) netG.to(opt.device) netG.eval() netG.load_state_dict(state_dict["netG"]) print(">=========================< load netG >==========================<") netM = Generator(opt, out_channels=1) netM.load_state_dict(state_dict["netM"]) print(">=========================< load netM >==========================<") netM.to(opt.device) netM.eval() netM.requires_grad_(False) test_dataloader = get_dataloader(opt, train=False, bd=False) inputs, targets = next(iter(test_dataloader)) inputs = inputs.to(opt.device) patterns = netG(inputs) patterns = netG.normalize_pattern(patterns) batch_masks = netM.threshold(netM(inputs)) bd_inputs = inputs + (patterns - inputs) * batch_masks bd_inputs = netG.denormalize_pattern(bd_inputs) * 255.0 bd_inputs = bd_inputs.detach().cpu().numpy() bd_inputs = np.clip(bd_inputs, 0, 255).astype(np.uint8).transpose((0, 2, 3, 1))怎么把这个bd_inputs的图片一张一张按PNG格式保存下来
你可以使用PIL库中的Image模块来保存bd_inputs的图像。以下是一个示例代码:
```python
from PIL import Image
# 保存bd_inputs中的每张图像
for i in range(len(bd_inputs)):
image = Image.fromarray(bd_inputs[i])
image.save(f"image_{i}.png")
```
这将保存每张图像为PNG格式,并以"image_0.png","image_1.png"等命名。你可以根据需要修改文件名或保存路径。
阅读全文
相关推荐
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![zip](https://img-home.csdnimg.cn/images/20241231045053.png)
![pdf](https://img-home.csdnimg.cn/images/20241231044930.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![rar](https://img-home.csdnimg.cn/images/20241231044955.png)