Создание бинарного классификатора в PyTorch
Вспомните, что небольшая нейронная сеть с одним линейным слоем и функцией сигмоиды — это бинарный классификатор. По своему принципу работы он аналогичен логистической регрессии.
Попрактикуйтесь в построении такой сети и интерпретации результатов классификатора.
Это упражнение является частью курса
Введение в глубокое обучение с 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)