mldas.explore.figures.plot_map_stage ¶
- mldas.explore.figures. plot_map_stage ( data , NeuralNet , loss_hist , datapath , k , img_size = 50 , channel_stride = 1 , sample_stride = 10 , overwrite = False , savefig = False ) [source] ¶
-
Plot stage of probability map.
- Parameters
-
-
data
numpy.ndarray -
Input raw measurements extracted from HDF5 file
-
NeuralNet
torch.nn.Module -
Initialized neural net architecture
-
loss_hist
numpy.ndarray -
History of loss values
-
datapath
str -
Path to saved data
-
k
int -
Index from loss/model history.
-
img_size
int -
Size of squared image
-
channel_stride
int -
Sliding interval in channel space
-
sample_stride
int -
Sliding interval in sample space
-
overwrite
bool -
Re-calculate probability map
-
savefig
bool -
Save figure as PNG file
-
data