File size: 2,552 Bytes
438eb35 13607ec 53aa80c 438eb35 13607ec 438eb35 13607ec 438eb35 13607ec 438eb35 53aa80c 438eb35 53aa80c 438eb35 53aa80c 438eb35 53aa80c 438eb35 13607ec 438eb35 13607ec 438eb35 13607ec 438eb35 13607ec 438eb35 13607ec 438eb35 13607ec 438eb35 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | {
"best_model": "LogReg_balanced",
"best_threshold": 0.6751170394620349,
"best_params": {
"clf__C": 0.1,
"clf__penalty": "l2",
"clf__solver": "lbfgs"
},
"metrics": {
"f1": 0.5816733067729084,
"recall": 0.6160337552742616,
"precision": 0.5509433962264151,
"roc_auc": 0.8373525516646647
},
"all_results": [
{
"model": "LogReg_balanced",
"metrics": {
"f1": 0.5816733067729084,
"recall": 0.6160337552742616,
"precision": 0.5509433962264151,
"roc_auc": 0.8373525516646647
},
"best_threshold": 0.6751170394620349,
"best_params": {
"clf__C": 0.1,
"clf__penalty": "l2",
"clf__solver": "lbfgs"
}
},
{
"model": "RF_balanced",
"metrics": {
"f1": 0.5408805031446541,
"recall": 0.5443037974683544,
"precision": 0.5375,
"roc_auc": 0.8203517201022514
},
"best_threshold": 0.41743914978839974,
"best_params": {
"clf__max_depth": 8,
"clf__min_samples_leaf": 8,
"clf__min_samples_split": 5,
"clf__n_estimators": 300
}
}
],
"baseline": {
"strategy": "most_frequent",
"majority_class": 0,
"f1": 0.0,
"recall": 0.0,
"precision": 0.0,
"roc_auc": 0.5
},
"features": {
"numerical": [
"age",
"revenu_mensuel",
"annees_dans_l_entreprise",
"annees_dans_le_poste_actuel",
"annees_depuis_la_derniere_promotion",
"distance_domicile_travail",
"nombre_participation_pee",
"note_evaluation_actuelle",
"note_evaluation_precedente",
"annees_depuis_le_changement_deposte",
"annee_experience_totale",
"augementation_salaire_precedente",
"nombre_experiences_precedentes",
"nb_formations_suivies",
"nombre_employee_sous_responsabilite",
"satisfaction_employee_environnement",
"satisfaction_employee_nature_travail",
"satisfaction_employee_equipe",
"satisfaction_employee_equilibre_pro_perso",
"augmentation_par_revenu",
"annee_sur_poste_par_experience",
"nb_formation_par_experience",
"score_moyen_satisfaction",
"dern_promo_par_experience",
"evolution_note"
],
"categorical": [
"genre",
"statut_marital",
"departement",
"poste",
"niveau_hierarchique_poste",
"niveau_education",
"domaine_etude",
"frequence_deplacement",
"heure_supplementaires"
]
},
"target": "a_quitte_l_entreprise"
} |