Extracting all three probabilities (neutral, bearish, bullish) rather than just the one with the highest probability
Hello,
I'm exploring your model and I would like to extract all three probabilities (neutral, bearish, bullish) because now I can only see the one with the highest probability. Can you please help me how to do it?
Thank you in advance!
Hi Jan,
Following the sample execution shown in the model card, the easiest way to achieve all 3 probs in the output is by setting the parameter 'return_all_scores = True' inside the TextClassificationPipeline. If you have any other questions, https://huggingface.co/docs/transformers/ contains the documentation where all the sample methods used can be investigated.
pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, max_length=64, truncation=True, padding = 'max_length', return_all_scores = True)
Happy coding!