You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I extracted my model using safetensors. In the original model I used cacluate_probabilities = True. When I loaded the model I wanted to predict the topics and the probabilities for a topic using the transform method.
The returned tuple was just the topic and the probability. Is there a way I can extract the probabilities for all the topics ? My goal is to get the top 10 topics for a document.
The text was updated successfully, but these errors were encountered:
It should have given you the probabilities if you loaded and saved with safetensors. Could you share your full code along with the version of BERTopic you used?
I extracted my model using safetensors. In the original model I used cacluate_probabilities = True. When I loaded the model I wanted to predict the topics and the probabilities for a topic using the transform method.
The returned tuple was just the topic and the probability. Is there a way I can extract the probabilities for all the topics ? My goal is to get the top 10 topics for a document.
The text was updated successfully, but these errors were encountered: