Binary Classification utilizando una red neuronal convolucional para perros y gatos
Este proyecto sirve de base para aprender a crear una red neuronal convolucional (CNN) y a estructurar un proyecto de ML para que sea reproducible y mantenible.
Motivación
El objetivo principal de este proyecto es entender la estructura del proyecto en lugar de lograr una alta precisión en las predicciones del modelo.
Lo que aprendí
A lo largo de este proyecto, he adquirido conocimientos en las siguientes áreas:
- Creación de una data pipeline simple utilizando scripts de Python y Makefiles para la orquestación.
- Entrenar un modelo CNN, guardarlo en disco y cargarlo para predicciones.
- Trabajar con archivos hdf5.
- Entrenamiento del modelo cargando datos del disco. Aunque los datos del proyecto caben cómodamente en memoria, incorporé la flexibilidad de entrenar el modelo por partes.
Cómo usarlo
Para explorar y ejecutar los scripts, puedes acceder al cuaderno de Google Colab proporcionado mediante el siguiente enlace:
CNN
Machine learning
ML Project
HD5F Files