Erro: BrokenPipeError: [Errno 32] Broken pipe - XGboost plot_tree | Classificação: melhorando o desempenho com XGBoost

# Configure o tamanho da figura e a resolução plt.figure(figsize=(15, 5), dpi=300) # Ajuste o tamanho e a resolução conforme necessário # Plotar a árvore xgb.plot_tree(xg_cl, num_trees=3, ax=plt.gca()) plt.show()

Oi, Hugo! Tudo bem contigo?

Para criar essa figura, você precisa seguir os seguintes passos:

Importar o XGBoost

import xgboost as xgb

Criar o modelo

xg_cl = xgb.XGBClassifier(objective='binary:logistic')
xg_cl.fit(X_treino, y_treino)
preds = xg_cl.predict(X_teste)

acuracia = accuracy_score(y_teste, preds)
print(f'Acuracia: {acuracia:.2f}')

Plotar a árvore

import matplotlib.pyplot as plt

# Configure o tamanho da figura e a resolução
plt.figure(figsize=(15, 5), dpi=300)  # Ajuste o tamanho e a resolução conforme necessário

# Plotar a árvore
xgb.plot_tree(xg_cl, num_trees=3, ax=plt.gca())
plt.show()

A figura exibida será assim: Insira aqui a descrição dessa imagem para ajudar na acessibilidade Tente novamente e se der algum erro me avise, tá?

Bons estudos :)

Mergulhe em programação

Importar o XGBoost

Criar o modelo

Plotar a árvore