小编Gol*_*rck的帖子

为 PyTorch 中的简单神经网络增加训练和测试精度

我知道这是一个原始问题但是我应该在我的代码中添加什么来输出神经网络的训练准确度以及损失,我查看了 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)

python neural-network python-3.x pytorch

5
推荐指数
1
解决办法
879
查看次数

标签 统计

neural-network ×1

python ×1

python-3.x ×1

pytorch ×1