Skip to content

Commit 85fc069

Browse files
committed
docs: add section for XGBClassifier
1 parent d7b2892 commit 85fc069

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

docs/source/get_started.rst

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,4 +153,13 @@ To compute QTE, we use "predict_qte" method. The confidence band is computed by
153153
:alt: QTE of adjusted estimator
154154
:height: 300px
155155
:width: 450px
156-
:align: center
156+
:align: center
157+
158+
You can use any model with "predict_proba" or "predict" method to adjust the distribution function estimation. For example, the following code use XGBoost classifier to estimate the conditional distribution.
159+
160+
.. code-block:: python
161+
162+
import xgboost as xgb
163+
estimator = dte_adj.AdjustedDistributionEstimator(xgb.XGBClassifier(), folds=3)
164+
estimator.fit(X, D, Y)
165+
cdf = estimator.predict(1, locations)

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "dte_adj"
7-
version = "0.1.2"
7+
version = "0.1.3"
88
description = "This is a Python library for a research paper 'Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction'"
99
readme = "README.md"
1010
requires-python = ">=3.10"

0 commit comments

Comments
 (0)