secondmind-labs / trieste

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

Try using step_callback to get gpflow training loss history #772

Closed uri-granta closed 12 months ago

uri-granta commented 1 year ago

Related issue(s)/PRs: #617

Summary

Investigate using the the the step callback mechaism to track training loss history for the gpflow scipy optimizer.

Having said that, it may be better to fix this in gpflow directly, though that would involve figuring out how (or whether) to handle interaction with other callbacks, as well as how to extend OptimizeResult.

Fully backwards compatible: yes

PR checklist

uri-granta commented 12 months ago

This will be covered by https://github.com/secondmind-labs/trieste/pull/774 and https://github.com/GPflow/GPflow/pull/2080