Skip to content

In SAGE, use directly calculated marginal mean as shortcut for empty coalition prediction (or at least featureless learner) #31

@jemus42

Description

@jemus42

Currently we actually take a trained learner and feed it a version of the task where every feature has been marginally sampled, which is, in essence, a very inefficient way to calculate mean(target) (in a regression or binary classif case).

Tried to validate one of the shapley assumptions for SAGE (sum(sage_values)baseline_loss - full_model_loss) and yeah, just using mean(target) lead to a much closer result, which isn't surprising.

Metadata

Metadata

Assignees

No one assigned

    Labels

    performanceDoesn't make it more correct but faster or less memory hungry

    Type

    No type

    Projects

    No projects

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions