Wi d dl Zd dlmZ d dlmZ d dej dedee e
f de
de de d ej fd
Z
y) N)Union
probabilitiesmin_probability
custom_labelstitlewidthheightreturnc
t |j dk7 r t d t |||kD dk( r t d t j ||k\ j }|| j
}t |t r|D cg c]# } t | dgg| j | | z % }
} |
D cg c]' } dj | dd D cg c] } | d c} ) }
} } |
D cg c] } t | dk r| n| dd d
z }
} n| j ,|r*|D
cg c] }
| j |
| j z }
}
ng }
|D ] }
| j |
}|rV|dd D cg c] }|d } }d |
d
dj | } t | dkD r| dd d
z n| } |
j | l|j ||
t! j" t! j$ ||
t' dt' dd d }|j) d| ddddt' dd d d ||t' d d d!" # |S c c} w c c} w c c} } w c c} w c c}
w c c}w )$ay Visualize the distribution of topic probabilities.
Arguments:
topic_model: A fitted BERTopic instance.
probabilities: An array of probability scores
min_probability: The minimum probability score to visualize.
All others are ignored.
custom_labels: If bool, whether to use custom topic labels that were defined using
`topic_model.set_topic_labels`.
If `str`, it uses labels from other aspects, e.g., "Aspect1".
title: Title of the plot.
width: The width of the figure.
height: The height of the figure.
Examples:
Make sure to fit the model before and only input the
probabilities of a single document:
```python
topic_model.visualize_distribution(probabilities[0])
```
Or if you want to save the resulting figure:
```python
fig = topic_model.visualize_distribution(probabilities[0])
fig.write_html("path/to/file.html")
```
zThis visualization cannot be used if you have set `calculate_probabilities` to False as it uses the topic probabilities of all topics. r zThere are no values where `min_probability` is higher than the probabilities that were supplied. Lower `min_probability` to prevent this error.N_ z... z Topic z: ( z#C8D2D7z#6E8484)colorr )r lineh)xymarker orientation Probabilitygffffff?g ?centertop Black)sizer )textr r xanchoryanchorfont simple_whitewhite Rockwell)bgcolor font_size font_family) xaxis_titler templater r
hoverlabel)lenshape
ValueErrornpargwhereflattentolist
isinstancestrtopic_aspects_joincustom_labels_ _outliers get_topicappendremovegoFigureBardict
update_layout) topic_modelr r r r r r
labels_idxvalstopiclabelsllabelidxwordswordfigs k/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/plotting/_distribution.pyvisualize_distributionrO s P = 1$
A
= 8 9:a?
_
]o =>FFHJ $ + + -D -%gqr^cCJ%&)C)CM)RSX)YYrrCIJa#((!BQ%8E!H89JJPVWu3u:?%cr
U0BBWW # # /MU_`c+,,S;3H3H-HI`` 0C ))#.E-22AY7Ta77#C5sxx.?@.1%j2ocr
U*5
e$ M# ./ 0
))
3
C ! g b0
2:N J] s8JW` 8s0 (I ?I I" I2 I &#I!- I&I)g Q?Fz%Topic Probability Distributioni iX )numpyr1 typingr plotly.graph_objects
graph_objectsr> ndarrayfloatboolr6 intr? rO rN rZ s ! #&+8g::g g s# g
g
g
g YYgrY