Skip to content

Commit

Permalink
cutting more words
Browse files Browse the repository at this point in the history
  • Loading branch information
CDonnerer committed Apr 15, 2023
1 parent 1723c7e commit e433d58
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -96,20 +96,19 @@ NGBoost performance comparison
``XGBDistribution`` follows the method shown in the `NGBoost`_ library, using natural
gradients to estimate the parameters of the distribution.

Below, we show a performance comparison of ``XGBDistribution`` with the `NGBoost`_
Below, we show a performance comparison of ``XGBDistribution`` and the `NGBoost`_
``NGBRegressor``, using the Boston Housing dataset, estimating normal distributions.
We note that while the performance of the two models is essentially identical (measured
on negative log-likelihood of a normal distribution and the RMSE), ``XGBDistribution``
is **30x faster** (timed on both fit and predict steps):
While the performance of the two models is essentially identical (measured on negative
log-likelihood of a normal distribution and the RMSE), ``XGBDistribution`` is
**30x faster** (timed on both fit and predict steps):

.. image:: https://raw.githubusercontent.com/CDonnerer/xgboost-distribution/main/imgs/performance_comparison.png
:align: center
:width: 600px
:alt: XGBDistribution vs NGBoost


Please see the `experiments page`_ in the documentation for detailed results across
various datasets.
Please see the `experiments page`_ for results across various datasets.


Full XGBoost features
Expand Down

0 comments on commit e433d58

Please sign in to comment.