Skip to content

About the model_category_mapping #631

Answered by fcakyon
jiangwenj02 asked this question in Q&A
Discussion options

You must be logged in to vote

You use category_mapping when you want to assign category names to category ids.

Assume your model predicts output category ids 0, 1, 2. Assume these category ids correspond to categories 'person', 'car', 'truck'. Then you set category_mapping={'0': 'person', '1': 'car', '2': 'truck'}.

Is it clear for you or do you still have questions?

Replies: 2 comments

Comment options

You must be logged in to vote
0 replies
Answer selected by fcakyon
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants
Converted from issue

This discussion was converted from issue #567 on September 30, 2022 18:35.