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

Learning with soft labels #6013

Open
sinchir0 opened this issue Jul 27, 2023 · 0 comments
Open

Learning with soft labels #6013

sinchir0 opened this issue Jul 27, 2023 · 0 comments
Labels

Comments

@sinchir0
Copy link

Summary

I would like LightGBM to be able to learn with the following soft labels.

example

y_train = np.array(
    [
        [0.3, 0.3, 0.4],
        [0.1, 0.7, 0.2],
        [0.2, 0.3, 0.5]
    ]
)

If the number of classes is 3, it is not impossible to use the custom objective function and copy the data 3 times to calculate loss. However, if the data is huge, it will consume a lot of memory and take a long time to calculate.

Motivation

For example, for a multiclass task, it is possible to have more detailed information in the objective.

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

No branches or pull requests

2 participants