Структура сети Rstudio Keras вызывает проблемы - PullRequest
Новая
       8

Структура сети Rstudio Keras вызывает проблемы

0 голосов
/ 06 февраля 2020

Я новичок в Keras и пытался использовать двунаправленный lstm для классификации каждой строки в тексте. Может кто-нибудь, пожалуйста, помогите получить правильную структуру сети. На более высоком уровне я пытаюсь использовать скрытое состояние каждой ячейки в качестве входных данных для проблемы классификации.

* Форма тензора данных: 2090 500

Форма тензора метки: 2090 500 5 * 

embedding_dim      <- 100

model <- keras_model_sequential() %>%
  layer_embedding(input_dim = max_words, output_dim = 32) %>%
  bidirectional(layer_lstm(units = 32)) %>%
  layer_dense(units = 5, activation = "softmax")



model %>% compile(
  optimizer = "adam",
  loss = "categorical_crossentropy",
  metrics = c("acc")
)

history <- model %>% fit(
  data_seq, labels_seq,
  epochs = 10,
  batch_size = 128,
  validation_split = 0.2
)
...