def get_default_readout(n_atom_basis):
    """Default setting for readout layers. Predicts only the energy of the system.

    Args:
        n_atom_basis (int): number of atomic basis. Necessary to match the dimensions of
            the linear layer.

    Returns:
        DEFAULT_READOUT (dict)
    """

    default_readout = {
        'energy': [
            {'name': 'linear', 'param': {'in_features': n_atom_basis,
                                         'out_features': int(n_atom_basis / 2)}},
            {'name': 'shifted_softplus', 'param': {}},
            {'name': 'linear', 'param': {'in_features': int(
                n_atom_basis / 2), 'out_features': 1}}
        ]
    }

    return default_readout