Merge pull request #47 from ziatdinovmax/master

Update examples
pycroscopy · Jun 18, 2021 · 379f746 · 379f746
2 parents d2f7cf3 + dcfd171
commit 379f746
Show file tree

Hide file tree

Showing 2 changed files with 158 additions and 179 deletions.
diff --git a/atomai/models/dklgp/dklgpr.py b/atomai/models/dklgp/dklgpr.py
@@ -7,6 +7,7 @@
 Created by Maxim Ziatdinov (email: maxim.ziatdinov@ai4microscopy.com)
 """
 
+import warnings
 from typing import Tuple, Type, Union
 
 import gpytorch
@@ -74,8 +75,11 @@ def _compute_posterior(self, X: torch.Tensor) -> gpytorch.distributions.Multivar
  """
  self.gp_model.eval()
  self.likelihood.eval()
- with torch.no_grad(), gpytorch.settings.use_toeplitz(False), gpytorch.settings.fast_pred_var():
- posterior = self.gp_model(X.to(self.device))
+ wrn = gpytorch.models.exact_gp.GPInputWarning
+ with warnings.catch_warnings():
+ warnings.filterwarnings("ignore", category=wrn)
+ with torch.no_grad(), gpytorch.settings.use_toeplitz(False), gpytorch.settings.fast_pred_var():
+ posterior = self.gp_model(X.to(self.device))
  return posterior
 
  def sample_from_posterior(self, X, num_samples: int = 1000) -> np.ndarray:

diff --git a/examples/notebooks/atomai_dkl_ferroic.ipynb b/examples/notebooks/atomai_dkl_ferroic.ipynb