Biased Model

[ParthMandaliya]

I plotted the confusion matrix with your model with test loader.

Model is biased it always outputs 0 regardless of input.

[issacchan26]

I plotted the confusion matrix with your model with test loader.

Model is biased it always outputs 0 regardless of input.

Thank you for reporting the issue, the model was trained in the preliminary test without finetuning. I did not sample and balance the data before, which cause low F1 score and too many False Positive. I will retrain the model and update the repo.

[Itsme1212]

hey can you provide the updated repo please !!! like i m getting the model biased and there is no balance in the data as well. Please employ all of them and make the updated repo