import conformal estimators Pt.2

thierrymoudiki · Aug 11, 2024 · fc7afef · fc7afef
1 parent 407d418
commit fc7afef
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 
 subprocess.check_call(['pip', 'install', 'Cython'])
 
-__version__ = "0.6.0"
+__version__ = "0.7.0"
 
 here = path.abspath(path.dirname(__file__))
 

diff --git a/unifiedbooster/gbdt_classification.py b/unifiedbooster/gbdt_classification.py
@@ -1,6 +1,6 @@
 from .gbdt import GBDT
 from sklearn.base import ClassifierMixin
-from .nonconformist import IcpClassifier, TcpClassifier    
+from .nonconformist import ClassifierAdapter, IcpClassifier, TcpClassifier, MarginErrFunc   
 
 try:
     from xgboost import XGBClassifier

diff --git a/unifiedbooster/gbdt_regression.py b/unifiedbooster/gbdt_regression.py
@@ -1,7 +1,7 @@
 from .gbdt import GBDT
 from sklearn.base import RegressorMixin
 from .predictioninterval import PredictionInterval
-from .nonconformist import IcpRegressor
+from .nonconformist import RegressorAdapter, IcpRegressor, RegressorNc, RegressorNormalizer, QuantileRegErrFunc
 
 try:
     from xgboost import XGBRegressor

diff --git a/unifiedbooster/nonconformist/__init__.py b/unifiedbooster/nonconformist/__init__.py
@@ -18,13 +18,17 @@
 )
 from .cp import IcpRegressor, TcpClassifier
 from .icp import IcpClassifier
-from .base import RegressorAdapter
+from .nc import ClassifierNc, MarginErrFunc
+from .base import RegressorAdapter, ClassifierAdapter
 
 __all__ = [
     "AbsErrorErrFunc",
+    "MarginErrFunc",
     "QuantileRegErrFunc",
     "RegressorAdapter",
+    "ClassifierAdapter",
     "RegressorNc",
+    "ClassifierNc", 
     "RegressorNormalizer",
     "IcpRegressor",
     "IcpClassifier",