This commit is contained in:
Starbeamrainbowlabs 2023-01-12 16:58:23 +00:00
parent 376eecc29f
commit e7c0328648
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -268,7 +268,7 @@ def plot_predictions(filepath, input_items, colormap, model):
filepath,
[
# input_tensor,
input_items[1], #label_colourmap
input_pair[1], #label_colourmap
prediction_colormap
],
figsize=(18, 14)