Reuse Keras models optimization

openvax / mhcflurry

Peptide-MHC I binding affinity prediction

http://openvax.github.io/mhcflurry/

Apache License 2.0

191 stars 57 forks source link

Reuse Keras models optimization #97

Closed timodonnell closed 7 years ago

timodonnell commented 7 years ago

This should give a substantial prediction speed improvement as identified by @rohanpai

Add borrow_cached_network() method to Class1NeuralNetwork
Use cached models at in Class1NeuralNetwork.predict
Set names for all neural network layers so the model JSON is identical for identical architectures

coveralls commented 7 years ago

Coverage decreased (-0.1%) to 75.224% when pulling 3505b640411afe15e115b3dc95d4efc29aaa05de on perf into 009905ddc48543b6567c7bb7d559141a4a2f5a4d on master.

rohanpai commented 7 years ago

LGTM

coveralls commented 7 years ago

Coverage decreased (-0.1%) to 75.224% when pulling 094435a9037e2fa9aa64b5842c15c950a5eff8d0 on perf into f7a89a407813d8a46c4f809e3c266192f91057ab on master.