This is a proyect about using Pre-trained models with the objective to predict Pneumonia on Xrays.
Use python3.11
mejores modelos loss (validation):
- ResNet50_ft_validation (0.0865)
- VGG16_freezed_validation (0.0807)
- VGG16_ft_validation (0.1025)
- ResNet50_freezed_validation (0.106)
mejores modelos Accuracy (validation):
- VGG16_freezzed_validation (0.9733)
- ResNet50_ft_validation (0.9685)
- VGG16_ft_validation (0.9637)
- ResNet50_freezed_validation (0.9618)
mejores modelos Recall (Validation):
- VGG16_freezed_validation (0.9835)
- ResNet50_freezed_validation (0.9848)
- VGG16_ft_validation (0.9674)
- Resnet50_ft_validation (0.9737)
Ranking
-
VGG16_freezed_validation: Se clasifica como el mejor modelo ya que tiene el menor valor de loss, la mayor precisión (accuracy) y el mayor recall.
-
ResNet50_ft_validation: Se clasifica en segundo lugar debido a su buen rendimiento en términos de loss y accuracy, aunque su recall es ligeramente menor que el primer modelo.
-
VGG16_ft_validation: Se posiciona en tercer lugar, ya que tiene un valor de loss y accuracy inferiores a los modelos anteriores, aunque su recall es aceptable.
-
ResNet50_freezed_validation: Se clasifica en último lugar, ya que tiene el mayor valor de loss y su rendimiento en términos de accuracy y recall es inferior a los otros modelos.