trainers.ForceMatching.predict

trainers.ForceMatching.predict#

ForceMatching.predict(dataset, params=None, batch_size=10)#

Computes predictions for a dataset.

Parameters:

  • dataset – Dictionary containing input data as numpy arrays. Can be, e.g., the whole testing split.

  • params – Parameters for the model. If None, uses the current parameters.

  • batch_size – Batch size for predictions.

Returns:

Returns all predictions of the model for the provided inputs.

Contents