Skip to content
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

Make class labels on the ConfusionMatrix publicly readable for custom charting support #7176

Open
IntegerMan opened this issue Jun 15, 2024 · 0 comments
Labels
enhancement New feature or request untriaged New issue has not been triaged

Comments

@IntegerMan
Copy link

Is your feature request related to a problem? Please describe.
I'm trying to build a library of data science charts for visualizing the results of ML.NET model training. One of those is a graphical confusion matrix pictured below (no offense to the wonderful built-in formatted table option present in ML.NET already).
image

Unfortunately, there's no public way of getting the names of classes for the matrix as ConfusionMatrix.PredictedClassesIndicators is internal. This means that class labels in multi-class classification charts must:

  1. Use an arbitrary label like "Class 1"
  2. Require the caller to specify an array of class names
  3. Call the public ConfusionMatrix.GetFormattedConfusionMatrix and parse the output to get the class names from the generated output.

Describe the solution you'd like
Either make PredictedClassesIndicators publicly gettable or provide a GetClassLabel(int classIndex): string method for accessing this information.

Describe alternatives you've considered
Right now I am requiring callers to provide an array of class names for multi-class classification. However, if this feature goes some time before it is implemented, I may decide to write code to parse the results of GetFormattedConfusionMatrix which does include those internal labels.

Additional context
My goal is for ML.NET to be as viable as SciKit-Learn in as many contexts as possible. Right now, I'm using ML.NET for my master's program and I'd much rather turn in a report with a graphical confusion matrix than a formatted text confusion matrix. Also, I'm running these charts in a Polyglot Notebook.

@IntegerMan IntegerMan added the enhancement New feature or request label Jun 15, 2024
@dotnet-policy-service dotnet-policy-service bot added the untriaged New issue has not been triaged label Jun 15, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request untriaged New issue has not been triaged
Projects
None yet
Development

No branches or pull requests

1 participant