from typing import Dict
from typing import Any


def get_dataset_config(dataset_name: str) -> Dict[str, Any]:
    """Get the dataset configuration

    Args:
        dataset_name (str): The name of the dataset to get the configuration for

    Returns:
        Dict[str, Any]: The dataset configuration
    """
    dataset_conf = {}
    if dataset_name == 'and':
        dataset_conf['inputs'] = [[0, 0], [0, 1], [1, 0], [1, 1]]
        dataset_conf['outputs'] = [[0], [0], [0], [1]]
    elif dataset_name == 'xor':
        dataset_conf['inputs'] = [[0, 0], [0, 1], [1, 0], [1, 1]]
        dataset_conf['outputs'] = [[0], [1], [1], [0]]
    elif dataset_name == 'class_example':
        dataset_conf['inputs'] = [0.05, 0.1]
        dataset_conf['desired_outputs'] = [0.01, 0.99]
        dataset_conf['weights'] = [[[0.15, 0.20, 0.35], [0.25, 0.30, 0.35]],
                                   [[0.40, 0.45, 0.60], [0.50, 0.55, 0.60]]]
    else:
        raise ValueError(f"Dataset name {dataset_name} not recognized.")

    return dataset_conf