From 481f403e1f34ae6b97676472234098046e11dc24 Mon Sep 17 00:00:00 2001 From: Dmitriy Serdyuk Date: Wed, 19 Sep 2018 17:51:32 -0400 Subject: [PATCH 1/2] Fix scatter without labels --- cortex/_lib/viz.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/cortex/_lib/viz.py b/cortex/_lib/viz.py index b080e47..da6d743 100644 --- a/cortex/_lib/viz.py +++ b/cortex/_lib/viz.py @@ -389,16 +389,15 @@ def save_heatmap(X, out_file=None, caption='', title='', image_id=0): def save_scatter(points, out_file=None, labels=None, caption='', title='', image_id=0): + names = data.DATA_HANDLER.get_label_names() if labels is not None: Y = (labels + 1.5).astype(int) + Y = Y - min(Y) + 1 + if len(names) != max(Y): + names = ['{}'.format(i + 1) for i in range(max(Y))] else: Y = None - names = data.DATA_HANDLER.get_label_names() - Y = Y - min(Y) + 1 - if len(names) != max(Y): - names = ['{}'.format(i + 1) for i in range(max(Y))] - visualizer.scatter( X=points, Y=Y, From c7cf45c4be25adedb78cf5bf4399f906b736737a Mon Sep 17 00:00:00 2001 From: Dmitriy Serdyuk Date: Thu, 20 Sep 2018 17:51:08 -0400 Subject: [PATCH 2/2] Fix names --- cortex/_lib/viz.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cortex/_lib/viz.py b/cortex/_lib/viz.py index da6d743..a0282b5 100644 --- a/cortex/_lib/viz.py +++ b/cortex/_lib/viz.py @@ -394,9 +394,10 @@ def save_scatter(points, out_file=None, labels=None, caption='', title='', Y = (labels + 1.5).astype(int) Y = Y - min(Y) + 1 if len(names) != max(Y): - names = ['{}'.format(i + 1) for i in range(max(Y))] + names = [names[y] for y in Y] else: Y = None + names = None visualizer.scatter( X=points,