Outputting Class Probabilities for Multiclass Models During Prediction in EasyRec
- Dominant language
- Python
- Stars
- 2.4k
- Forks
- 383
- PR merge metrics
- No merged PRs in 30d
Description
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.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reproducing the prediction command with -Doutput_cols='probs string,probs_y double,y bigint' and inspect how the prediction command parses output_cols and multiclass probability outputs. Done means identifying the accepted type or format for probs and documenting or validating a working command.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100