matteo/constellationnet
matteo
/
constellationnet
1
0
Fork 0
Code Issues Pull Requests Releases Activity

Add larger set of colors for plot

This commit is contained in:
Mattéo Delabre 2019-12-15 23:52:57 -05:00
parent 722c02ef24
commit e4457400a6
Signed by: matteo
GPG Key ID: AE3FBD02DC583ABB
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
constellation/util.py
View File

@ -1,5 +1,7 @@
import torch import torch
import matplotlib import matplotlib
from matplotlib.colors import ListedColormap
import seaborn
def get_random_messages(count, order): def get_random_messages(count, order):
@ -61,7 +63,7 @@ def plot_constellation(
ax.grid() ax.grid()
order = len(constellation) order = len(constellation)
color_map = matplotlib.cm.Dark2 color_map = ListedColormap(seaborn.color_palette('husl', n_colors=order))
color_norm = matplotlib.colors.BoundaryNorm(range(order + 1), order) color_norm = matplotlib.colors.BoundaryNorm(range(order + 1), order)
# Extend axes symmetrically around zero so that they fit data # Extend axes symmetrically around zero so that they fit data