在Pytorch中使用样本权重(sample_weight)的正确方法

P = [0.1,0.2,0.4,0.3]Q = [0,0,1,0]loss = -Q * np.log(P)

P = [0.1,0.2,0.4,0.3]Q = [0,0,sample_weight,0]loss_samle_weight = -Q * np.log(P)

train_data = np.load(open('train_data.npy','rb'))train_labels = []for i in range(8):  train_labels += [i] *100train_labels = np.array(train_labels)train_labels = to_categorical(train_labels).astype("float32")sample_1 = [random.random() for i in range(len(train_data))]for i in range(len(train_data)):  floor = i / 100  train_labels[i][floor] = sample_1[i]train_data = torch.from_numpy(train_data) train_labels = torch.from_numpy(train_labels) dataset = dataf.TensorDataset(train_data,train_labels) trainloader = dataf.DataLoader(dataset, batch_size=batch_size, shuffle=True)

def my_loss(outputs, targets):    output2 = outputs - torch.max(outputs, 1, True)[0]    P = torch.exp(output2) / torch.sum(torch.exp(output2), 1,True) + 1e-10    loss = -torch.mean(targets * torch.log(P))    return loss