Closed Thunfischpirat closed 1 year ago
When we updated the models to return class embeddings instead of class probabilities so that we could create the t-SNE embeddings for task II.3, we forgot to adjust the training loop of train_model()
. in line 111, we now applied the softmax to the model output. This fixes the issues. #12
Somehow the neural network model's performance during training doesn't surpass 10% anymore. Why is that the case?