Browse Source

[MNT] Add more details about cnn_gp

tags/v0.3.2
shihy 2 years ago
parent
commit
29b1df9fde
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      learnware/specification/image/cnn_gp.py

+ 4
- 0
learnware/specification/image/cnn_gp.py View File

@@ -7,6 +7,10 @@ import math

__all__ = ("NNGPKernel", "Conv2d", "ReLU", "Sequential", "ConvKP", "NonlinKP")
"""
With this package, we are able to accurately and efficiently compute the kernel matrix corresponding to the NNGP during the search phase.

Github Repository: https://github.com/cambridge-mlg/cnn-gp

References: [1] A. Garriga-Alonso, L. Aitchison, and C. E. Rasmussen, ‘Deep Convolutional Networks as shallow Gaussian Processes’, in International Conference on Learning Representations, 2019.
"""



Loading…
Cancel
Save