secondmind-labs / trieste

A Bayesian optimization toolbox built on TensorFlow
Apache License 2.0
212 stars 42 forks source link

Expose optimize results for trainable models #774

Closed uri-granta closed 12 months ago

uri-granta commented 12 months ago

Related issue(s)/PRs: #617

Summary

Allow trainable models to return an optimization result and provide a backwards compatible method to make it accessible in the models. This currently includes OptimizeResult for gpflow models using the SciPy optimizer, keras.callback.History for keras models using the keras fit method, and lists of results for model stacks.

Note that exposing this in the model makes more sense than exposing it in the optimizer wrappers, as these are essentially an implementation detail and behave very differently between e.g. gpflow and keras.

Fully backwards compatible: yes

PR checklist