1. Learn
    2. /
  2. Курси
    3. /
  3. Глибоке навчання для тексту з PyTorch
Connected

Вправа

Створення моделі LSTM для тексту

У PyBooks команда постійно прагне покращити користувацький досвід, використовуючи найновіші технології. У межах цього підходу вам доручили важливе завдання. Команда хоче, щоб ви дослідили потенціал ще одного потужного інструменту — LSTM, який добре вловлює складніші закономірності в даних. Ви працюєте з тим самим набором даних Newsgroup, і мета незмінна: класифікувати новинні статті на три окремі категорії:

rec.autos, sci.med та comp.graphics.

Наступні пакети вже завантажено для вас: torch, nn, optim.

Інструкції

100 XP
  • Налаштуйте модель LSTM, доповнивши шари LSTM і лінійний шар необхідними параметрами.
  • Ініціалізуйте модель із потрібними параметрами.
  • Навчіть модель LSTM: обнуляйте градієнти та пропускайте вхідні дані X_train_seq через модель.
  • Обчисліть функцію втрат на основі передбачених outputs і правильних міток.