Skip to content

Methods to only compute diagonal of covariance functions #2344

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

Merged
merged 3 commits into from
Jun 26, 2017

Conversation

bwengals
Copy link
Contributor

  • __call__ of covariance functions have new arg in signature: cov(X, Z, diag=True)
    • diag=True computes and returns only the diagonal
    • default is False
  • adds tests

this should help speed up some of the sparse approximations

pymc3/gp/cov.py Outdated
R"""
Evaluate the kernel/covariance function.

Parameters
----------
X : The training inputs to the kernel.
Z : The optional prediction set of inputs the kernel. If Z is None, Z = X.
daig: Return only the diagonal of the covariance function. Default is False.
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Typo: daig

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

those daig typos

@fonnesbeck fonnesbeck merged commit 4bba2f0 into pymc-devs:master Jun 26, 2017
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