Merge pull request #7 from lokeshkumarn/patch-1

Update model.R
kasaai · Apr 1, 2019 · 46a0733 · 46a0733
2 parents 4e3ebac + 8532a32
commit 46a0733
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/analysis/model.R b/analysis/model.R
@@ -3,7 +3,8 @@ dt_model <- function() {
   company_code_input <- layer_input(shape = 1, name = "company_input")
   company_code_embedding <- company_code_input %>%
     layer_embedding(200, 49, name = "company_code_embedding") %>%
-    layer_flatten()
+    layer_flatten()%>%
+    layer_repeat_vector(9)
 
   encoded <- ay_seq_input %>%
     layer_masking(-99) %>%
@@ -12,7 +13,8 @@ dt_model <- function() {
   decoded <- encoded %>%
     layer_repeat_vector(9) %>%
     layer_gru(128, return_sequences = TRUE, dropout = 0.2, recurrent_dropout = 0.2) %>%
-    layer_lambda(f = function(x) layer_concatenate(list(x, k_repeat(company_code_embedding, 9))))
+    layer_lambda(f = function(x) layer_concatenate(list(x, company_code_embedding)))  
+    #layer_lambda(f = function(x) layer_concatenate(list(x, k_repeat(company_code_embedding, 9))))
 
   case_reserves_output <- decoded %>%
     time_distributed(layer_dense(units = 64, activation = "relu")) %>%