SOMPlots
SOMPlots functions.
- susi.SOMPlots.plot_estimation_map(estimation_map: ndarray, cbar_label: str = 'Variable in unit', cmap: str = 'viridis', fontsize: int = 20) Axes [source]
Plot estimation map.
- Parameters:
estimation_map (np.ndarray) – Estimation map of the size (n_rows, n_columns)
cbar_label (str, optional) – Label of the colorbar, by default “Variable in unit”
cmap (str, optional (default=”viridis”)) – Colormap
fontsize (int, optional (default=20)) – Fontsize of the labels
- Returns:
ax – Plot axis
- Return type:
pyplot.axis
- susi.SOMPlots.plot_nbh_dist_weight_matrix(som, it_frac: float = 0.1) Axes [source]
Plot neighborhood distance weight matrix in 3D.
- Parameters:
som (susi.SOMClustering or related) – Trained (un)supervised SOM
it_frac (float, optional (default=0.1)) – Fraction of som.n_iter_unsupervised for the plot state.
- Returns:
ax – Plot axis
- Return type:
pyplot.axis
- susi.SOMPlots.plot_som_histogram(bmu_list: List[Tuple[int, int]], n_rows: int, n_columns: int, n_datapoints_cbar: int = 5, fontsize: int = 22) Axes [source]
Plot 2D Histogram of SOM.
Plot 2D Histogram with one bin for each SOM node. The content of one bin is the number of datapoints matched to the specific node.
- Parameters:
bmu_list (list of (int, int) tuples) – Position of best matching units (row, column) for each datapoint
n_rows (int) – Number of rows for the SOM grid
n_columns (int) – Number of columns for the SOM grid
n_datapoints_cbar (int, optional (default=5)) – Maximum number of datapoints shown on the colorbar
fontsize (int, optional (default=22)) – Fontsize of the labels
- Returns:
ax – Plot axis
- Return type:
pyplot.axis
- susi.SOMPlots.plot_umatrix(u_matrix: ndarray, n_rows: int, n_colums: int, cmap: str = 'Greys', fontsize: int = 18) Axes [source]
Plot u-matrix.
- Parameters:
u_matrix (np.ndarray) – U-matrix containing the distances between all nodes of the unsupervised SOM. Shape = (n_rows*2-1, n_columns*2-1)
n_rows (int) – Number of rows for the SOM grid
n_columns (int) – Number of columns for the SOM grid
cmap (str, optional (default=”Greys)) – Colormap
fontsize (int, optional (default=18)) – Fontsize of the labels
- Returns:
ax – Plot axis
- Return type:
pyplot.axis