Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import torch
- import torch.nn as nn
- n_in_neurons = 5
- n_hidden_neurons = 3
- n_out_neurons = 1
- net = nn.Sequential(
- nn.Linear(n_in_neurons, n_hidden_neurons),
- nn.ReLU(),
- nn.Linear(n_hidden_neurons, n_out_neurons),
- nn.BCEWithLogitsLoss()
- )
- x = torch.Tensor([-0.23, -0.2, 0.31, -0.9, 0.2])
- y = torch.Tensor([0,])
- y_hat = net.forward(x) # прогоните x через нейронную сеть
- loss = nn.BCEWithLogitsLoss() # инициализируйте функцию потерь бинарной кросс-энтропии
- print(loss(y_hat, y))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement