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

Outputting Class Probabilities for Multiclass Models During Prediction in EasyRec #436

Open
KKKatherineeee opened this issue Dec 8, 2023 · 1 comment

Comments

@KKKatherineeee
Copy link

KKKatherineeee commented Dec 8, 2023

Hello EasyRec Team,

I am using EasyRec for a multiclass prediction task and I have encountered an issue with the data type of the 'probs' output when running the prediction command. The error message indicates that 'probs' is not of type string. Here is the command I am using:

-Doutput_cols='probs string,probs_y double,y bigint'

I would greatly appreciate it if you could provide guidance on the correct way to specify the data type for 'probs' in the context of the prediction command. Is there a specific format or type that EasyRec expects for probability outputs?

Thank you for your assistance.

@chengmengli06
Copy link
Collaborator

probs is actually a vector of floats, on ODPS, currently only probs_y(the probs corresponding to y) could be saved, to save probs, a slight modification of the source code is required to convert probs into a comma separated string, such as: '0.1,0.2,0.6,0.1'.

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

2 participants