Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify Python getting started example (#8153)

Browse Source 
Load data set via `sklearn` rather than a local file path.
This commit is contained in:
Ravi Makhija 2022-08-11 04:42:09 -04:00 committed by GitHub
parent d868126c39
commit 20d1bba1bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
doc/get_started.rst
View File

@ -19,16 +19,18 @@ Python
.. code-block:: python .. code-block:: python
import xgboost as xgb from xgboost import XGBClassifier
# read in data # read data
dtrain = xgb.DMatrix('demo/data/agaricus.txt.train') from sklearn.datasets import load_iris
dtest = xgb.DMatrix('demo/data/agaricus.txt.test') from sklearn.model_selection import train_test_split
# specify parameters via map data = load_iris()
param = {'max_depth':2, 'eta':1, 'objective':'binary:logistic' } X_train, X_test, y_train, y_test = train_test_split(data['data'], data['target'], test_size=.2)
num_round = 2 # create model instance
bst = xgb.train(param, dtrain, num_round) bst = XGBClassifier(n_estimators=2, max_depth=2, learning_rate=1, objective='binary:logistic')
# make prediction # fit model
preds = bst.predict(dtest) bst.fit(X_train, y_train)
# make predictions
preds = bst.predict(X_test)
*** ***
R R