segmentation_models
segmentation_models copied to clipboard
Weights loading and evaluation
Hello everybody,
I trained my model using weighted dice loss + categorical focal loss and saved the weights in the file weights.h5, now I have to load my weights in another notebook for prediction and post processing so i have to call the model and compile it with the same loss I used in taining, do I have to use the same class weights or use other class weights for testing ( in fact for training i calculated class weights with a sklearn function) please can someone answer my question so I can be assured I am doing the right thing ?

Thanks in advance.