Skip to content

Commit

Permalink
define clearly class_names and labels
Browse files Browse the repository at this point in the history
  • Loading branch information
cpranav93 committed Nov 14, 2023
1 parent 589762f commit 4e53208
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -183,8 +183,8 @@ num_samples = 500 # The number of samples to generate for the LIME explainer.
Which of your model's classes do you want an explanation for?

```python
class_names= [class_a, class_b] # The list of class names.
labels = np.argsort(class_names) # The list of labels for different classes.
class_names= [class_a, class_b] # String representation of the different classes of interest
labels = np.argsort(class_names) # Numerical representation of the different classes of interest for the model
```

Run using the XAI method of your choice, for example LIME with the following additional arguments:
Expand Down

0 comments on commit 4e53208

Please sign in to comment.