DTM
¶
Visualize topics over time
Parameters:
Name | Type | Description | Default |
---|---|---|---|
topic_model |
A fitted BERTopic instance. |
required | |
topics_over_time |
DataFrame
|
The topics you would like to be visualized with the corresponding topic representation |
required |
top_n_topics |
int
|
To visualize the most frequent topics instead of all |
None
|
topics |
List[int]
|
Select which topics you would like to be visualized |
None
|
normalize_frequency |
bool
|
Whether to normalize each topic's frequency individually |
False
|
custom_labels |
Union[bool, str]
|
If bool, whether to use custom topic labels that were defined using
|
False
|
title |
str
|
Title of the plot. |
'<b>Topics over Time</b>'
|
width |
int
|
The width of the figure. |
1250
|
height |
int
|
The height of the figure. |
450
|
Returns:
Type | Description |
---|---|
Figure
|
A plotly.graph_objects.Figure including all traces |
Examples:
To visualize the topics over time, simply run:
topics_over_time = topic_model.topics_over_time(docs, timestamps)
topic_model.visualize_topics_over_time(topics_over_time)
Or if you want to save the resulting figure:
fig = topic_model.visualize_topics_over_time(topics_over_time)
fig.write_html("path/to/file.html")
Source code in bertopic\plotting\_topics_over_time.py
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
|