1. Learn
    2. /
  2. Cursuri
    3. /
  3. Metode de ansamblu în Python
Connected

exercițiu

Prezicerea comestibilității ciupercilor

Acum că ai explorat datele, este momentul să construiești un prim model pentru a prezice comestibilitatea ciupercilor.

Setul de date îți este disponibil ca mushrooms. Deoarece atât caracteristicile, cât și ținta sunt categoriale, acestea au fost transformate în variabile binare „dummy" pentru tine.

Să începem cu Naive Bayes (folosind GaussianNB din scikit-learn) și să vedem cum se descurcă acest algoritm pe această problemă.

Instrucțiuni

100 XP
  • Instanțiază un clasificator GaussianNB numit clf_nb.
  • Antrenează clf_nb pe datele de antrenament X_train și y_train.
  • Calculează predicțiile pe setul de testare. Aceste predicții vor fi folosite pentru a evalua performanța folosind scorul de acuratețe.