ML ZOOMCAMP 2025 - Module 8

 Module 8: Neural networks and Deep learning

In module we learned how to build an image classification model using PyTorch and transfer learning. We used a clothing dataset.

What was covered

  • Introduction to PyTorch for deep learning
  • Loading and preprocessing image data
  • Using pre-trained models (MobileNetV2)
  • Understanding convolutional neural networks (CNNs)
  • Transfer learning: adapting pre-trained models
  • Hyperparameter tuning: learning rate optimization
  • Model checkpointing: saving the best model
  • Adding more layers to improve performance
  • Dropout regularization to prevent overfitting
  • Data augmentation for better generalization
  • Training the final model
  • Using the model for predictions
  • Exporting models to ONNX format




Comments

Popular posts from this blog

ML ZOOMCAMP 2025 - Module 1

ML ZOOMCAMP 2025 - Module 3

ML ZOOMCAMP 2025 - Module 6