Skip to content

Commit

Permalink
documentation update 1
Browse files Browse the repository at this point in the history
  • Loading branch information
MHuberFaust committed Jul 3, 2018
1 parent c48790e commit ab2eb56
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions dariah_topics/visualization.py
Expand Up @@ -60,11 +60,14 @@
class PlotDocumentTopics:
"""
Class to visualize document-topic matrix.
Args:
document_topics (pandas.Dataframe): document-topics matrix. Produced by postprocessing.show_document_topics
"""
def __init__(self, document_topics):
self.document_topics = document_topics



def static_heatmap(self, figsize=(1000 / 96, 600 / 96), dpi=None,
labels_fontsize=13, cmap='Blues', ticks_fontsize=12,
xlabel='Document', ylabel='Topic', xticks_bottom=0.1,
Expand Down

0 comments on commit ab2eb56

Please sign in to comment.