0

I have a datasets that looks like this: Training (Class 0: 471, Class 1: 986) Testing (Class 0: 177, Class 1: 246. I split my data as 80% for training and 20% for validation. I know that is an imbalanced dataset, and I have tried Class_weight but the problem remains. I have retrained my Baseline CNN and I always have a result like as attached in the picture. Could someone help me? My Result after training

Test12345
  • 1,625
  • 1
  • 12
  • 21

1 Answers1

0

I faced a similar problem while classifying events in 5 imbalanced categories. I found this loss function that implement a weighted categorical cross-entropy: https://gist.github.com/noparade/aaa8584e6e90ad64936e333e4e08ca5f Combined with the Nadam optimizer, it allowed me to get over 95% true positive for all my categories.

charlesnadeau
  • 26
  • 1
  • 2
  • 1