sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-07-02 11:24:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

embeddings: change title rendering; make even moar widererer

Browse source 
We need to see that parallel coordinates  plot in detail
This commit is contained in:
Starbeamrainbowlabs 2022-09-23 18:56:39 +01:00
parent b4ddb24589
commit d59de41ebb
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -12,7 +12,7 @@ def vis_embeddings(filepath_output, features):
).fit(features)
px = 1 / plt.rcParams['figure.dpi'] # matplotlib sizes are in inches :-( :-( :-(
width = 3840
width = 8000
height = 768
plt.rc("font", size=20)
@ -24,7 +24,7 @@ def vis_embeddings(filepath_output, features):
umap.plot.points(dimreducer,
ax=axes["A"]
)
plt.title(f"UMAP Dimensionality Reduction", fontsize=20)
axes["A"].set_title(f"UMAP Dimensionality Reduction", fontsize=20)
# 2: Parallel coordinates
dataframe = pandas.DataFrame(features)
@ -39,7 +39,7 @@ def vis_embeddings(filepath_output, features):
sort_labels=True
)
plt.title(f"Parallel coordinates plot", fontsize=20)
axes["B"].set_title(f"Parallel coordinates plot", fontsize=20)
plt.suptitle(f"RainfallContrastive embeddings | rainfall | E2 ConvNeXt | {len(features)} items", fontsize=28, weight="bold")
plt.savefig(filepath_output)