experiments.experiment_utils.batch_predictions_custom_regime

batch_predictions_custom_regime(model, solution, test_data, **kwargs)

Run model forward pass in batches for the custom regime.

Parameters:

• model – A model object with a .predict(theta,data) method

• solution – Model weights to set before making the forward pass

• test_data – The input data to the model to batch up

Returns:

y_pred, the combined predictions in a flattened array