Skip to content

ANN neural network model to detect whether a user stays in the bank or not. This algorithm is part of a course I took, but I adapted and improved some steps.

Notifications You must be signed in to change notification settings

Marcos14Almeida/ANN_model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

ANN_model

Descrição do Projeto

Modelo de rede neural ANN para detectar se um usuário permanece ou não no banco. Esse algoritmo faz parte de um curso que fiz, mas adaptei e melhorei algumas etapas.

Como usar o Projeto

Através do uso de Python, eu uso um dataset com informações a respeito de clientes de um banco. A partir disso, eu filtro o dataset em train_set e test_set aplico feature scaling e o método de rede neural ANN no train_set com a biblioteca Keras para obter uma previsão dos resultados. Modificando os hiperparâmetros e observando as métricas de Confusion Matrix e Accuracy, o algoritmo foi aperfeiçoado para melhores resultados.

Como Executar o projeto

Baixe os arquivos e com o uso de algum compilador python com suporte para Keras, Tensorflow, Numpy, Pandas e Sklearn instalados execute o código do arquivo ".py".

OU rode o arquivo ".ipynb" do Google Collab. *Não esqueça de adicionar o dataset no diretório de arquivos para funcionar.

Resultado

Confusion Matrix =

[[43 10]

[11 14]]

Accuracy = 0.730 F1-Score = 0.687

About

ANN neural network model to detect whether a user stays in the bank or not. This algorithm is part of a course I took, but I adapted and improved some steps.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published