1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Nhập môn Deep Learning với Keras
Connected

Bài tập

Xây dựng mô hình LSTM của bạn

Bạn đã chuẩn bị xong các chuỗi văn bản. Đến lúc xây dựng mô hình LSTM!

Hãy nhớ là mỗi chuỗi của bạn có 4 từ; mô hình sẽ được huấn luyện trên 3 từ đầu tiên của mỗi chuỗi để dự đoán từ thứ 4. Bạn sẽ dùng một lớp Embedding để học cách biến các từ thành các vector có ý nghĩa. Những vector này sau đó sẽ được đưa vào một lớp LSTM đơn giản. Đầu ra là một lớp Dense với số neuron bằng số từ trong vốn từ và dùng kích hoạt softmax. Lý do là vì chúng ta muốn chọn ra từ tiếp theo có xác suất cao nhất trong tất cả các khả năng.

Kích thước vốn từ (số lượng từ duy nhất) được lưu trong vocab_size.

Hướng dẫn

100 XP
  • Import các lớp Embedding, LSTM và Dense từ các lớp của tensorflow.keras.
  • Thêm một lớp Embedding() với kích thước vốn từ, biến các từ thành vector 8 số và nhận các chuỗi có độ dài 3.
  • Thêm một lớp LSTM() với 32 neuron.
  • Thêm một lớp ẩn Dense() 32 neuron và một lớp đầu ra vocab_size neuron với softmax.