НачатьНачать бесплатно

Построение прогнозирующей РНС

Пришло время построить первую рекуррентную сеть! Это будет модель типа «последовательность в вектор», состоящая из слоя РНС с двумя слоями и hidden_size равным 32. После слоя РНС простой линейный слой преобразует выходные данные в одно прогнозируемое значение.

Следующие импорты уже выполнены за вас:

import torch
import torch.nn as nn

Это упражнение является частью курса

Глубокое обучение на PyTorch: средний уровень

Посмотреть курс

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

class Net(nn.Module):
    def __init__(self):
        super().__init__()
        # Define RNN layer
        self.rnn = ____(
            ____,
            ____,
            ____,
            ____,
        )
        self.fc = nn.Linear(32, 1)
Редактировать и запускать код