Este repositório contém código Python para treinar e avaliar modelos de redes neurais convolucionais (CNNs) usando TensorFlow e Keras. O objetivo principal é classificar dígitos escritos à mão usando o conjunto de dados MNIST. Descrição do Código
O código inclui as seguintes etapas:
-
Importação de Bibliotecas e Carregamento de Dados: Importação das bibliotecas necessárias, carregamento dos dados do conjunto de dados MNIST e pré-processamento das imagens.
-
Definição e Treinamento do Modelo: Definição da arquitetura do modelo CNN usando a API Sequencial do Keras, compilação do modelo com otimizador 'adam' e função de perda 'sparse_categorical_crossentropy', e treinamento do modelo com os dados de treinamento.
-
Avaliação do Modelo e Matriz de Confusão: Avaliação do modelo treinado usando os dados de teste e geração de uma matriz de confusão para avaliar o desempenho do modelo.
-
TensorBoard: Utilização do TensorBoard para visualização e monitoramento do treinamento, incluindo métricas como acurácia, sensibilidade, F1 Score e área sob a curva (AUC).
Para utilizar este código, siga estas etapas:
- Clone o repositório para o seu ambiente local.
- Execute o código Python em um ambiente compatível com TensorFlow e Keras.
- Analise os resultados do treinamento, incluindo métricas de desempenho, usando o TensorBoard.
- Python
- TensorFlow
- Keras
- Matplotlib
- Seaborn
- Pandas
Contribuições são bem-vindas! Sinta-se à vontade para abrir problemas ou enviar solicitações de pull com melhorias ou correções.
Este projeto foi feito para estudo junto com a DIO.me, o python foi utilizado o Coleb e dúvidas foram tiradas do ChatGPT (IA).
Wagner Nogueira