# `TrainBatchPredict`

### *class* capymoa.ocl.evaluation.events.TrainBatchPredict[[source]](https://github.com/adaptive-machine-learning/CapyMOA/blob/3e255b1/src/capymoa/ocl/evaluation/events.py#L41)

Bases: [`TrainTaskBegin`](capymoa.ocl.evaluation.events.TrainTaskBegin.md#capymoa.ocl.evaluation.events.TrainTaskBegin)

After predicting on a training batch, but before training on it.

#### \_\_init_\_(train_task: [int](https://docs.python.org/3/builtins/functions.html#int), global_step: [int](https://docs.python.org/3/builtins/functions.html#int), batch: [int](https://docs.python.org/3/builtins/functions.html#int), x: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor), y: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor), y_hat: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor)) → [None](https://docs.python.org/3/builtins/constants.html#None)

#### batch *: [int](https://docs.python.org/3/builtins/functions.html#int)*

Batch ID within the current task stream.

#### global_step *: [int](https://docs.python.org/3/builtins/functions.html#int)*

A monotonically increasing integer that counts training and evaluation steps.

#### train_task *: [int](https://docs.python.org/3/builtins/functions.html#int)*

The ID of the training task that has begun.

#### x *: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor)*

The input batch.

#### y *: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor)*

The target batch.

#### y_hat *: [Tensor](https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor)*

The predicted batch.