[madlib] branch master updated: DL: Fix transfer learning dev-check test

[kh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

khannaekta pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/madlib.git


The following commit(s) were added to refs/heads/master by this push:
     new eb77116  DL: Fix transfer learning dev-check test
eb77116 is described below

commit eb771166ef706437edf154e5f70c4b945acbffb3
Author: Orhan Kislal <ok...@apache.org>
AuthorDate: Thu Dec 19 17:10:33 2019 -0500

    DL: Fix transfer learning dev-check test
---
 .../deep_learning/test/madlib_keras_transfer_learning.sql_in | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in b/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
index 6a7ec1f..3c970a5 100644
--- a/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
+++ b/src/ports/postgres/modules/deep_learning/test/madlib_keras_transfer_learning.sql_in
@@ -292,6 +292,16 @@ SELECT load_model_selection_table(
     ]
 );
 
+DROP TABLE if exists iris_multiple_model, iris_multiple_model_summary, iris_multiple_model_info;
+SELECT setseed(0);
+SELECT madlib_keras_fit_multiple_model(
+  'iris_data_packed',
+  'iris_multiple_model',
+  'mst_table',
+  3,
+  FALSE, NULL, 1
+);
+
 UPDATE iris_model_arch
 SET model_weights = (SELECT model_weights FROM iris_multiple_model WHERE mst_key=1)
 WHERE model_id = 1;
@@ -308,7 +318,7 @@ SELECT madlib_keras_fit_multiple_model(
   'iris_multiple_model',
   'mst_table',
   3,
-  FALSE
+  FALSE, NULL, 1
 );
 
 SELECT assert(