我知道这是一个原始问题但是我应该在我的代码中添加什么来输出神经网络的训练准确度以及损失,我查看了 PyTorch 教程,它们展示了如何在图像分类中添加训练/测试准确度但是我不知道如何在我的简单 XOR 求解 NN 中做到这一点,以下是代码:
# Step 1: importing our dependencies
import torch
from torch.autograd import Variable
import numpy as np
# Our data
x = Variable(torch.Tensor([[0, 0, 1], [0, 1, 1], [1, 0, 1], [0, 1, 0], [1, 0, 0],
[1, 1, 1], [0, 0, 0]]))
y = Variable(torch.Tensor([[0], [1], [1], [1], [1], [0], [0]]))
# Step 2: building our class model
class NeuralNetwork(torch.nn.Module):
def __init__(self):
super(NeuralNetwork, self).__init__()
self.linear_ij = torch.nn.Linear(3, 4)
self.linear_jk = torch.nn.Linear(4, 1)
def …Run Code Online (Sandbox Code Playgroud)