1. 06 Apr, 2021 1 commit
    • Jeremiah Liu's avatar
      Disable temperature scaling during training. · fab47e9e
      Jeremiah Liu authored
      For the `GaussianProcessClassificationHead`, the temperature scaling needs to be disabled during training to avoid unexpected modification to the learning rate, which harms model quality. (Unfortunately, this seems to require adding `training` to the `call` method).
      
      Also set the default of `gp_cov_ridge_penalty` in `RandomFeatureGaussianProcess` to 1 to be consistent with that in the `GaussianProcessClassificationHead`.
      
      PiperOrigin-RevId: 366917075
      fab47e9e
  2. 05 Apr, 2021 1 commit
    • Jeremiah Liu's avatar
      Adds `call()` method to `GaussianProcessClassificationHead`. · 2ae06c8a
      Jeremiah Liu authored
      This change allows `GaussianProcessClassificationHead` to output only the predictive logits during training and evaluation (instead of outputting a tuple  `(logits, covmat)`). The goal is to make the layer more compatible with `SentencePredictionTask` and Keras' `model.fit()` API.
      
      PiperOrigin-RevId: 366891298
      2ae06c8a
  3. 01 Apr, 2021 3 commits
  4. 24 Mar, 2021 1 commit
  5. 22 Mar, 2021 3 commits
  6. 20 Mar, 2021 1 commit
  7. 17 Mar, 2021 1 commit
  8. 16 Mar, 2021 1 commit
  9. 12 Mar, 2021 2 commits
  10. 10 Mar, 2021 1 commit
  11. 09 Mar, 2021 1 commit
  12. 08 Mar, 2021 1 commit
  13. 04 Mar, 2021 1 commit
  14. 02 Mar, 2021 1 commit
  15. 26 Feb, 2021 1 commit
  16. 23 Feb, 2021 1 commit
  17. 16 Feb, 2021 2 commits
  18. 28 Jan, 2021 1 commit
  19. 19 Jan, 2021 1 commit
  20. 14 Jan, 2021 2 commits
  21. 12 Jan, 2021 1 commit
  22. 08 Jan, 2021 1 commit
  23. 05 Jan, 2021 1 commit
  24. 21 Dec, 2020 1 commit
  25. 10 Dec, 2020 1 commit
  26. 09 Dec, 2020 1 commit
  27. 01 Dec, 2020 1 commit
  28. 25 Nov, 2020 1 commit
  29. 16 Nov, 2020 1 commit
  30. 11 Nov, 2020 1 commit
  31. 26 Oct, 2020 1 commit
  32. 16 Oct, 2020 1 commit
  33. 07 Oct, 2020 1 commit