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

[New Feature] extending SoftmaxLoss.py to accept for class probabilities in target #3228

Open
i-plusplus opened this issue Feb 11, 2025 · 0 comments

Comments

@i-plusplus
Copy link

SoftmaxLoss loss currently accepts class labels[0-n) in the target.

One important use case is to fit a model on observed data with no one category but a mix of multiple classes. Changing this class to handle probabilities along with classes will cater to this use case.

Code change also seems simple. I can do the code change, if this feature is accepted.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant