datasetprediction

DatasetPredictionEnvironment(environment, batch_size)

Bases: ActiveEnvironment

Dataset prediction environment.

Initialize the dataset prediction environment.

Parameters:

Name	Type	Description	Default
environment	Dataset	The dataset to use for prediction.	required
batch_size	int	The batch size of the prediction.	required

Source code in src/flowcean/polars/environments/datasetprediction.py

def __init__(
    self,
    environment: Dataset,
    batch_size: int,
) -> None:
    """Initialize the dataset prediction environment.

    Args:
        environment: The dataset to use for prediction.
        batch_size: The batch size of the prediction.
    """
    super().__init__()
    self.environment = environment
    self.batch_size = batch_size