federicafiorentini / Employers-Attrition-Analysis

Machine Learning Project

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Employers' attrition

University Project

Cattura

Per evidenziare i fattori principali che portano un dipendente ad abbandonare la propria azienda è stato impostato un problema di classificazione utilizzando un dataset fornito dalla piattaforma Kaggle. Il dataset è riferito a 1470 dipendenti della società multinazionale IBM, su cui sono state osservate 35 variabili.

Sono stati sviluppati diversi modelli di Machine Learning quali: Decision Tree, NeuralNetwork (NNET), Logistic Regression, Support Vector Machine (SVM), Stochastic Gradient Boosting (GBM) e Random Forest.