Skip to content

OutputConversionTable

Bases: OutputConversion[Table, TabularDataset]

The output conversion for a neural network, defines the output parameters for the neural network.

Source code in src/safeds/ml/nn/_output_conversion_table.py 
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class OutputConversionTable(OutputConversion[Table, TabularDataset]):
    """The output conversion for a neural network, defines the output parameters for the neural network."""

    def __init__(self, prediction_name: str = "prediction") -> None:
        """
        Define the output parameters for the neural network in the output conversion.

        Parameters
        ----------
        prediction_name:
            The name of the new column where the prediction will be stored.
        """
        self._prediction_name = prediction_name

    def _data_conversion(self, input_data: Table, output_data: Tensor, **kwargs: Any) -> TabularDataset:  # noqa: ARG002
        return input_data.add_columns([Column(self._prediction_name, output_data.tolist())]).to_tabular_dataset(
            self._prediction_name,
        )