Hey
@RBR, for your latest comment, the last Dense layer in the model you are looking at is untrained, so it’s very unlikely that it will 100% reflect your label (the green component).
What you are seeing from the Dense layer is its output distribution - how probable it thinks each output is. When you do inference on it, it will automatically pick the highest value (through ArgMax) which in this case would be label nr 1 (as opposed to 0).
I might have misunderstood your concern though, so let me know if I didn’t answer it properly 
Also, what would you want to edit with the labels? They are automatically set based on the datatype and CSV which you are using.
As for the earlier question in this thread - how to get the output distribution rather than the single output value:
We have a feature coming in soon where you can toggle pre-and-post-processing on or off, and without the post-processing you would get the output distribution.
Until then though, you can do something like this:
import tensorflow as tf
import os
from PIL import Image
import numpy as np
from tensorflow import keras
from keras import backend as K
mapping = {0: "No", 1: "Yes"}
path_to_model = "Brain_tumor/model"
def load_image(path):
image = Image.open(path)
image = np.array(image, dtype=np.float32)
image = np.expand_dims(image, axis=0)
return image
model = keras.models.load_model(path_to_model)
#Load some image
image = load_image("C:/Users/lundb/Documents/PerceptiLabs/Default/Code/Brain_tumor/dataset/no/1 no.jpeg")
#Loads the model
model = keras.models.load_model(path_to_model)
#Take the input to the post-processing layer as the output of the model
inp = model.input
input_to_postprocessing = model.layers[-1].input
model_func = K.function(inp, input_to_postprocessing)
#Makes some predictions and catogirizes them
prediction1 = model_func(image)
print(prediction1) #This prints the output distribution
print(mapping[np.asarray(prediction1).argmax()]) #This prints the value you would get if you use postprocessing
Hope that helps! 