Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Training on energy labels #128

Merged
merged 126 commits into from
May 14, 2020
Merged

Training on energy labels #128

merged 126 commits into from
May 14, 2020

Conversation

jonpvandermause
Copy link
Collaborator

@jonpvandermause jonpvandermause commented Dec 15, 2019

GPs can now be trained on both force and energy labels. The following modules have been modified to allow this:

  • gp.py: the update_db method now takes an optional energy argument.
  • gp_algebra.py: the covariance matrix and kernel vectors now include force/energy and energy/energy kernels.
  • mgp_en.py: now includes kernels with respect to energy labels, allowing GPs trained on forces and energies to be mapped.

Some notes:

  • mgp.py has not been updated; it will only work if the underlying GP was trained on force labels only.
  • hyperparameter optimization considers only the force/force block of the covariance matrix. This can be generalized by implementing gradients of the force/energy and energy/energy kernels with respect to the hyperparameters (currently only gradients of the force/force kernel are implemented)

@jonpvandermause jonpvandermause changed the title WIP: energy GPs Training on energy labels May 13, 2020
@jonpvandermause jonpvandermause changed the base branch from master to development May 13, 2020 19:22
@jonpvandermause jonpvandermause merged commit 94455b1 into development May 14, 2020
@jonpvandermause jonpvandermause deleted the feature/jon/gp-energy branch May 15, 2020 19:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants