1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Machine Learning with Tree-Based Models in Python

Connected

cvičení

Trénování klasifikátoru AdaBoost

Teď, když máš klasifikátor ada vytvořený, je čas ho natrénovat. Také vypočítáš pravděpodobnosti příslušnosti k pozitivní třídě v testovací sadě. Postup je následující:

Jakmile je klasifikátor ada natrénovaný, zavolej metodu .predict_proba() s parametrem X_test a vyber tyto pravděpodobnosti tak, že vezmeš všechny hodnoty z druhého sloupce:

ada.predict_proba(X_test)[:,1]

Dataset Indian Liver je již zpracovaný a rozdělený na 80 % trénovací a 20 % testovací sadu. Matice příznaků X_train a X_test, stejně jako pole popisků y_train a y_test jsou dostupné v tvém pracovním prostoru. Navíc jsme načetli také instanciovaný model ada z předchozího cvičení.

Pokyny

100 XP
  • Natrénuj ada na trénovací sadě.

  • Vyhodnoť pravděpodobnosti příslušnosti k pozitivní třídě v testovací sadě.