experiments.experiment_utils.batch_predictions

batch_predictions(model, solution, X_test, **kwargs)

Run model forward pass in batches.

Parameters:

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

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

• X_test – The features to batch up

Returns:

y_pred, the combined predictions in a flattened array