tree b182dc879ae7455b259e8483305872fb49597431
parent bc0f13285087e31c82ee423fdb9d5eea61471599
author Nikhil Kak <nkak@pivotal.io> 1583272433 -0800
committer kaknikhil <kaknikhil@users.noreply.github.com> 1583518689 -0800

DL: improvements to "serializing weights with image count"

JIRA: MADLIB-1416

We need to add the image count to the model weights to get the final
state for fit_transition.
Previously we would use np.concatenate to join the image count and model
weights together. A cleaner and faster way is to append image
count to the weights list and get rid of the np.concatenate call.

Ran unit test `test_serialize_image_nd_weights_valid_output` with large
model_weights `[np.array([1]*100000000), np.array([1]*100000000)]` to
confirm the speed improvements.

test_serialize_image_nd_weights_valid_output with old code and large
weights: 17.8 s
test_serialize_image_nd_weights_valid_output with new code and large
weights: 16.4 s