How can I adjust the figure size of an XGBoost plot tree?

To adjust the figure size of an XGBoost plot tree, you can use the matplotlib library, which XGBoost uses for plotting. Here's how you can do it:

1. Import Necessary Libraries:

- Make sure you have xgboost and matplotlib installed. If not, you can install them using pip: pip install xgboost matplotlib.

- Import the required modules in your Python script.

2. Generate the XGBoost Plot:

- Use the xgboost.plot_tree() function to generate the plot. This function returns a matplotlib.axes.Axes object.

3. Adjust the Figure Size:

- Before calling plt.show(), use matplotlib.pyplot.figure() to create a figure object and set its size using the figsize parameter. Then, pass the ax object to the plot_tree function.

4. Example Code:

import xgboost as xgb
import matplotlib.pyplot as plt

# Sample XGBoost model (replace with your actual model)
data = [[1, 2], [3, 4], [5, 6]]
labels = [0, 1, 0]
dmatrix = xgb.DMatrix(data, label=labels)
params = {'objective': 'binary:logistic', 'n_estimators': 1, 'max_depth': 2}
model = xgb.train(params, dmatrix)

# Set the figure size
fig = plt.figure(figsize=(20, 10)) # Adjust width and height as needed
ax = fig.add_subplot(111)
# Plot the tree with the specified figure size
xgb.plot_tree(model, ax=ax)
plt.show()


5. Explanation:

- figsize=(20, 10) sets the width to 20 inches and the height to 10 inches. You can adjust these values to fit your needs.

- fig.add_subplot(111) creates a subplot within the figure, which is necessary for the plot_tree function to draw on.

- By passing the ax object to xgb.plot_tree(), you ensure that the plot is drawn on the specified figure and axes.

By following these steps, you can effectively control the size of your XGBoost plot tree, making it more readable and suitable for your presentation or analysis.

Learn more about this