sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

ai vis: make parallel coordinates wider

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-16 18:51:49 +01:00
parent a70794e661
commit a552cc4dad
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -12,17 +12,17 @@ def vis_embeddings(filepath_output, features):
).fit(features) ).fit(features)
px = 1 / plt.rcParams['figure.dpi'] # matplotlib sizes are in inches :-( :-( :-( px = 1 / plt.rcParams['figure.dpi'] # matplotlib sizes are in inches :-( :-( :-(
width = 1920 width = 3840
height = 768 height = 768
plt.rc("font", size=20) plt.rc("font", size=20)
plt.rc("font", family="Ubuntu") plt.rc("font", family="Ubuntu")
figure = plt.figure(figsize=(width*px, height*px)) figure, axes = plt.subplot_mosaic("ABBBB", figsize=(width*px, height*px))
figure.add_subplot(1, 2, 1) # figure.add_subplot(1, 2, 1)
# 1: UMAP # 1: UMAP
umap.plot.points(dimreducer, umap.plot.points(dimreducer,
ax=figure.get_axes()[0] ax=axes["A"]
) )
plt.title(f"UMAP Dimensionality Reduction", fontsize=20) plt.title(f"UMAP Dimensionality Reduction", fontsize=20)
@ -34,7 +34,7 @@ def vis_embeddings(filepath_output, features):
pandas.plotting.parallel_coordinates( pandas.plotting.parallel_coordinates(
dataframe, dataframe,
"Label", "Label",
ax=figure.get_axes()[1], ax=axes["B"],
use_columns=False, use_columns=False,
axvlines=False, axvlines=False,
sort_labels=True sort_labels=True