Treinar o classificador AdaBoost
Agora que você instanciou o classificador AdaBoost ada
, é hora de treiná-lo. Você também preverá as probabilidades de obter a classe positiva no conjunto de teste. Isso pode ser feito da seguinte forma:
Depois que o classificador ada
for treinado, chame o método .predict_proba()
passando X_test
como parâmetro e extraia essas probabilidades cortando todos os valores na segunda coluna da seguinte forma:
ada.predict_proba(X_test)[:,1]
O conjunto de dados Indian Liver Patient é processado para você e dividido em 80% de treinamento e 20% de teste. As matrizes de recursos X_train
e X_test
, bem como as matrizes de rótulos y_train
e y_test
estão disponíveis em seu espaço de trabalho. Além disso, também carregamos o
modelo instanciado ada
do exercício anterior.
Este exercício faz parte do curso
Aprendizado de máquina com modelos baseados em árvores em Python
Instruções de exercício
Ajuste
ada
ao conjunto de treinamento.Avalie as probabilidades de obter a classe positiva no conjunto de teste.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Fit ada to the training set
____
# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]