Skip to content

Commit e0cf73c

Browse files
committed
Extend docstring and add reference
1 parent 65c326a commit e0cf73c

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

autoemulate/emulators/conformal.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,21 @@ class Conformal(Emulator):
8181
8282
This class wraps a base emulator to provide conformal prediction intervals with
8383
guaranteed frequentist coverage.
84+
85+
Both standard split conformal and Conformalized Quantile Regression (CQR) methods
86+
are supported.
87+
88+
Conformalized Quantile Regression (CQR) is defaultly implemented with two neural net
89+
quantile regressors predicting lower and upper quantiles, followed by a calibration
90+
step to ensure valid coverage. Note the _fit_quantile_regressors method can be
91+
overridden to implement custom quantile regressors.
92+
93+
References
94+
----------
95+
- Romano, Y., Patterson, E., & Candes, E. (2019). Conformalized Quantile Regression.
96+
In Advances in Neural Information Processing Systems (Vol. 32).
97+
https://papers.nips.cc/paper/8613-conformalized-quantile-regression.pdf
98+
8499
"""
85100

86101
supports_uq = True

0 commit comments

Comments
 (0)