def get_onnx_input_data(dataset):
    """Transforms the Dataset object to the ONNX model input.

    Args:
        dataset (Dataset): The data set to use for creating ONNX input.

    Returns:
        (dict) The dictionary defining the ONNX model input.
    """
    return {
        column: dataset.X[column].values.reshape(-1, 1)
        for column in dataset.X.columns
    }