-
Notifications
You must be signed in to change notification settings - Fork 195
[WIP] Added a confusion matrix plot generator. #274
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
because ours much more beautiful! ;) See discussion end of #242 so it was considered ! But I would try to add the class names in the example |
We agreed that we can keep like this (with own function that shows a bit more information than scikitlearn) depending on how strong @agramfort objects to it :D |
I won't strongly object I am just more lazy than you ;)
… |
confusion_mat = confusion_matrix(y_true, y_pred) | ||
|
||
# add class labels | ||
label_dict = valid_set.datasets[0].windows.event_id.items() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
label_dict = valid_set.datasets[0].windows.event_id.items() | |
# label_dict is class_name : str -> i_class : int | |
label_dict = valid_set.datasets[0].windows.event_id.items() |
And maybe just add same code also for trialwise example, why not? |
Great work! |
refering to #242
@Ann-KathrinKiessner @dcwil @HenrikBons