1. Learn
  2. /
  3. Cursuri
  4. /
  5. Analiza de supraviețuire în Python

Connected

exercițiu

Studiu privind fluctuația angajaților

Recrutarea de noi angajați implică costuri de angajare și de formare profesională. Vrei să prezici cât timp vor rămâne angajații actuali. Acest exercițiu se concentrează pe primii pași necesari pentru a face predicții.

Ai un DataFrame numit employees. Acesta conține date despre 1470 de angajați (plecați și prezenți) și răspunsurile lor la un sondaj. Sondajul acoperă următoarele dimensiuni:

  • environment_satisfaction
  • job_satisfaction
  • relationship_satisfaction
  • work_life_balance

În plus, years_at_company reprezintă durata în care angajații au lucrat în companie, iar attrition indică dacă angajatul a plecat (1 dacă a plecat, 0 în caz contrar).

În consolă sunt afișate câteva rânduri din set. Clasa CoxPHFitter este importată pentru tine din pachetul lifelines.

Instrucțiuni 1/2

undefined XP
    1
    2
  • Instanțiază un obiect CoxPHFitter.
  • Antrenează obiectul model CoxPHFitter pe datele tale, folosind toate celelalte coloane.