kaknikhil commented on a change in pull request #443: DL: Add training for multiple models URL: https://github.com/apache/madlib/pull/443#discussion_r326747650
########## File path: src/ports/postgres/modules/deep_learning/test/madlib_keras_model_selection.sql_in ########## @@ -158,3 +133,180 @@ SELECT assert( ) FROM mst_table; +m4_changequote(`<!', `!>') +m4_ifdef(<!__POSTGRESQL__!>, <!!>, <! +-- Multiple models test +-- Prepare model selection table with three rows +DROP TABLE IF EXISTS mst_table, mst_table_summary; +SELECT load_model_selection_table( + 'iris_model_arch', + 'mst_table', + ARRAY[1], + ARRAY[ + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.01)', metrics=['accuracy']$$, + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.001)', metrics=['accuracy']$$, + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.0001)', metrics=['accuracy']$$ + ], + ARRAY[ + $$batch_size=50, epochs=1$$ + ] +); +-- Prepare model selection table with only one row +DROP TABLE IF EXISTS mst_table_1row, mst_table_1row_summary; +SELECT load_model_selection_table( + 'iris_model_arch', + 'mst_table_1row', + ARRAY[1], + ARRAY[ + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.01)', metrics=['accuracy']$$ + ], + ARRAY[ + $$batch_size=16, epochs=1$$ + ] +); +-- Prepare model selection table with four rows +DROP TABLE IF EXISTS mst_table_4row, mst_table_4row_summary; +SELECT load_model_selection_table( + 'iris_model_arch', + 'mst_table_4row', + ARRAY[1], + ARRAY[ + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.01)', metrics=['accuracy']$$, + $$loss='categorical_crossentropy', optimizer='Adam(lr=0.001)', metrics=['accuracy']$$ + ], + ARRAY[ + $$batch_size=16, epochs=1$$, + $$batch_size=32, epochs=1$$ + ] +); + +-- Test when number of configs(3) equals number of segments(3) Review comment: This test case depends on the assumption that madlib is installed on a 3 segment host. Can we instead get this number from gp_segment_configuration and then create the configs based on that ? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] With regards, Apache Git Services
