models icon indicating copy to clipboard operation
models copied to clipboard

[FEA] Add post_block layer to MultiClassClassificationTask class

Open rnyak opened this issue 3 years ago • 1 comments

🚀 Feature request

Currently we get logits as outputs if we train a MultiClassClassificationTask() model. It'd be useful to add an optional post-block layer in the MultiClassClassificationTask class and set it to softmax layer as default. That way users can generate predict probas as well.

rnyak avatar May 27 '22 16:05 rnyak

in the prediction mode we can make the softmax as default to get the probas.

rnyak avatar May 31 '22 15:05 rnyak