Skip to content
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ def count_labels(ds):
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ def count_labels(ds):
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def main():
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def main():
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
2 changes: 1 addition & 1 deletion examples/interpretability/mp_stageattn_mimic4_interpret.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def main():
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -120,7 +120,7 @@ def main():
"ig_gim": IntegratedGradientGIM(model),
"deeplift": DeepLift(model, use_embeddings=True),
"gim": GIM(model),
# "chefer": CheferRelevance(model),
"chefer": CheferRelevance(model),
"shap": ShapExplainer(model, use_embeddings=True),
"lime": LimeExplainer(model, use_embeddings=True, n_samples=200),
}
Expand Down
Loading