Add missing kwargs to optimize_scale

LLNL / MuyGPyS

A fast, pure python implementation of the MuyGPs Gaussian process realization and training algorithm.

Other

25 stars 11 forks source link

Add missing kwargs to optimize_scale #235

Open igoumiri opened 5 months ago

igoumiri commented 5 months ago

I ran into an error when trying to optimize the scale for a hierarchical model where I wasn't able to pass batch_features to the kernel. This fixes it.

Example usage:

muygps = muygps.optimize_scale(batch_pairwise_diffs, batch_nn_targets, batch_features=batch_features)