Wi d dl Z d dlZd dlmZmZ d dlmZ d dl m
Z
ddee dz de de dee e
f de
d e d
e d e d ej fd
Zy) N)ListUnion)
make_subplotstopics top_n_topicsn_words
custom_labelstitlewidthheight autoscalereturnc t j g d } | j }
|
j |
j dk7 ddf }
| t | }nO|'t
|
j j d| }n&t
|
j j dd }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 } }nJ| j ,|r*|D cg c] } | j | | j z } } n|D cg c] } d | } } d}t t! j" t | |z }t% ||d d
|dkD rd|z nd| }d}d}|D ] } | j' | D cg c]
\ }}|dz c}}d| ddd }| j' | D cg c] \ }}| c}}d| ddd }|j) t+ j, ||dt/ | || |r[t | dkD rdt | dz
dz z }t | dkD r+|j1 t3 |dz
t | z ||k( rd}|dz
}|dz
} |j5 d d | d d d t3 d d!" d#|dz |dkD r||z n|d$z t3 d%d&d'( ) |j7 d*+ |j1 d*+ |S c c} w c c}w c c}}
w c c}w c c} w c c} w c c}}w c c}}w ),a Visualize a barchart of selected topics.
Arguments:
topic_model: A fitted BERTopic instance.
topics: A selection of topics to visualize.
top_n_topics: Only select the top n most frequent topics.
n_words: Number of words to show in a topic
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 each figure.
height: The height of each figure.
autoscale: Whether to automatically calculate the height of the figures to fit the whole bar text
Returns:
fig: A plotly figure
Examples:
To visualize the barchart of selected topics
simply run:
```python
topic_model.visualize_barchart()
```
Or if you want to save the resulting figure:
```python
fig = topic_model.visualize_barchart()
fig.write_html("path/to/file.html")
```
)z#D55E00z#0072B2z#CC79A7z#E69F00z#56B4E9z#009E73z#F0E442Nr _ z...zTopic Fg? g?)rowscols shared_xaxeshorizontal_spacingvertical_spacingsubplot_titlesz h)xy orientation marker_color)rowcol )size)tickfont plotly_whiteg ?centertop Black)r) color)textr xanchoryanchorfontg?white Rockwell)bgcolor font_size font_family)template
showlegendr
r r
hoverlabelT)showgrid) itertoolscycleget_topic_freqlocTopiclistsortedto_list
isinstancestrtopic_aspects_joinlencustom_labels_ _outliersintnpceilr get_topic add_tracegoBarnext update_yaxesdict
update_layout update_xaxes) topic_modelr r r r r
r r r
colorsfreq_dftopicr labelslabelcolumnsr figr" columnwordr wordsscorescoress g/home/sietch6/trending-topics-pipeline/venv/lib/python3.12/site-packages/bertopic/plotting/_barchart.pyvisualize_barchartrh s \ __ j
kF ((*Gkk'--2-q01G
f !
--/
>?
--/!45 -%ouvfkCJ-.1K1KM1Z[`1aavvUcd6#((&!*#EE!H#EFddXfgu3u:?%cr
U8JJgg # # /MaghX]+44U[=R=R5RShh8>?uF5'*??G rwws6{W,- .D
'+axtQ%
C C
F ,7,A,A%,HIqI(7STXVXTXY(3(=(=e(DEHAu%EhwOPTRTPTU
FFVu#DL Q
5zB E
R2 555zA~ $Vc\c%j4P*Q R W F 1HC aKF+0 g b0
ai $qv}fsl2:N
d#d# J w#Edgh? JEs< 9(M(M? M M M#M9 M ,M% M+ M)N FzTopic Word Scoresr% r% F)r? numpyrO typingr r plotly.graph_objects
graph_objectsrS plotly.subplotsr rN boolrH Figurerh rg rt s ! )
$&++{ I { { {
s# {
{ {
{ { YY{rs