ComeçarComece gratuitamente

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

Ver Curso

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 = ____.____(____)[____]
Editar e executar código