diff --git a/pyLDAvis/_prepare.py b/pyLDAvis/_prepare.py index d50c1881..30bee0a0 100644 --- a/pyLDAvis/_prepare.py +++ b/pyLDAvis/_prepare.py @@ -163,7 +163,7 @@ def js_TSNE(distributions, **kwargs): tsne : array, shape (`n_dists`, 2) """ dist_matrix = squareform(pdist(distributions, metric=_jensen_shannon)) - model = TSNE(n_components=2, random_state=0, metric='precomputed', **kwargs) + model = TSNE(n_components=2, random_state=0, metric='precomputed', perplexity=min(len(dist_matrix)-1, 30), **kwargs) return model.fit_transform(dist_matrix) @@ -468,4 +468,4 @@ def to_dict(self): 'topic.order': self.topic_order} def to_json(self): - return json.dumps(self.to_dict(), cls=NumPyEncoder) \ No newline at end of file + return json.dumps(self.to_dict(), cls=NumPyEncoder)