Побудова бінарного класифікатора в PyTorch
Згадайте, що мала нейронна мережа з одним лінійним шаром, після якого йде сигмоїдна функція, є бінарним класифікатором. Вона працює так само, як логістична регресія.
Попрактикуйтеся у створенні такої невеликої мережі та інтерпретації виходу класифікатора.
Ця вправа є частиною курсу
Вступ до Deep Learning з PyTorch
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
import torch
import torch.nn as nn
input_tensor = torch.Tensor([[3, 4, 6, 2, 3, 6, 8, 9]])
# Implement a small neural network for binary classification
model = nn.Sequential(
nn.____(____),
nn.____()
)
output = model(input_tensor)
print(output)