Machine Learning with NumPy, pandas, scikit-learn, and More/

...

/

XGBoost Basics

XGBoost Basics

Learn about the basics of using XGBoost.

We'll cover the following...
  • Time to Code!

    • Chapter Goals:

    • Learn about the XGBoost data matrix
    • Train a Booster object in XGBoost

    A. Basic data structures

    The basic data structure for XGBoost is the DMatrix, which represents a data matrix. The DMatrix can be constructed from NumPy arrays.

    The code below creates DMatrix objects with and without labels.

    Press + to interact
    Python 3.5
    data = np.array([
      [1.2, 3.3, 1.4],
      [5.1, 2.2, 6.6]])
    import xgboost as xgb
    dmat1 = xgb.DMatrix(data)
    labels = np.array([0, 1])
    dmat2 = xgb.DMatrix(data, label=labels)

    The DMatrix object can be used for training and using a ...

    Ask