1. Learn
  2. /
  3. Cursuri
  4. /
  5. Învățare supravegheată cu scikit-learn

Connected

exercițiu

Împărțirea în seturi de antrenament/testare + calcularea acurateței

E momentul să exersezi împărțirea datelor în seturi de antrenament și testare folosind setul de date churn_df!

Array-urile NumPy au fost deja create pentru tine: X conține caracteristicile, iar y conține variabila țintă.

Instrucțiuni

100 XP
  • Importă train_test_split din sklearn.model_selection.
  • Împarte X și y în seturi de antrenament și testare, setând test_size la 20%, random_state la 42 și asigurând că proporțiile etichetelor țintă reflectă distribuția din setul de date original.
  • Antrenează modelul knn pe datele de antrenament.
  • Calculează și afișează acuratețea modelului pe datele de testare.