Skip to content

KernelSHAP raises a warning if output dim is 1 and task is regression #226

Open
@jklaise

Description

@jklaise

This could be confusing for the user as simple regression is a valid use case:

if not self._explainer.vector_out:
logger.warning(
"Predictor returned a scalar value. Ensure the output represents a probability or decision score "
"as opposed to a classification label!"
)

Same applies to the open TreeSHAP PR (#206).

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions