Unify _predict() signatures, add missing "diag" to predict_f()

cics-nd / gptorch

Gaussian processes with PyTorch

MIT License

30 stars 8 forks source link

Closed sdatkinson closed 5 years ago

sdatkinson commented 5 years ago

Signature: _predict(self, input_new: torch.Tensor, diag=True)
GPModel.predict_f(), predict_y, predict_y_samples take care of np->torch conversion via a decorator
Expose missing diag kwarg for GPModel.predict_f()
Update demos