Get predicted standard deviation for Gaussian process regression

vicrsp Dataiku DSS Core Designer, Registered Posts: 3 ✭✭✭

Hi, I built a Gaussian Process regressor as a custom model in a Visual Analysis, and deployed it to the Flow and an API. For each model prediction, I need the predicted mean and standard deviation to be returned, but I couldn't find a way to do it. This is achieved by calling the model's predict method with the parameter return_std=True.

Is it possible to display the standard deviation for each prediction when evaluating the deployed model in the flow? Moreover, how can I make an API call return the standard deviation?

Setup Info
      Help me…