1. Learn
    2. /
  2. Cursuri
    3. /
  3. Preprocesare pentru Machine Learning în Python
Connected

exercițiu

Antrenarea Naive Bayes cu selecție de caracteristici

Acum vei rula din nou modelul de clasificare a textului Naive Bayes pe care l-ai folosit la finalul Capitolului 3, de data aceasta cu opțiunile de selecție din exercițiul anterior: coloanele title și category_desc din setul de date volunteer.

Instrucțiuni

100 XP
  • Aplică train_test_split() pe vectorul de text filtered_text, etichetele y (adică etichetele category_desc) și transmite setul y parametrului stratify, deoarece distribuția claselor este inegală.
  • Antrenează modelul Naive Bayes nb pe X_train și y_train.
  • Calculează acuratețea modelului nb pe setul de testare.