sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

embeddings: fix title; remove colourmap

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 17:08:04 +01:00
parent 4fd852d782
commit 5778fc51f7
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/lib/vis/embeddings.py
View file

@ -20,7 +20,6 @@ def vis_embeddings(filepath_output, features):
# 1: UMAP
umap.plot.points(dimreducer,
color_key_cmap="brg", # color_key_cmap="jet",
ax=figure.get_axes()[0]
)
plt.title(f"UMAP Dimensionality Reduction", fontsize=20)
@ -41,5 +40,5 @@ def vis_embeddings(filepath_output, features):
plt.title(f"Parallel coordinates plot", fontsize=20)
plt.suptitle(f"ContrastiveE1 embeddings | ResNetV2 | {len(features)} items", fontsize=28, weight="bold")
plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold")
plt.savefig(filepath_output)