diff --git a/glest/__init__.py b/glest/__init__.py index 83549e0..688ed7d 100644 --- a/glest/__init__.py +++ b/glest/__init__.py @@ -1,3 +1,3 @@ -__version__ = "0.0.1-alpha.1" +__version__ = "0.0.1" from .core import GLEstimator, GLEstimatorCV, Partitioner diff --git a/glest/core.py b/glest/core.py index e90ecf1..7475c28 100644 --- a/glest/core.py +++ b/glest/core.py @@ -18,11 +18,11 @@ ) from .plot import grouping_diagram from sklearn.utils.validation import indexable -from sklearn.base import clone +from sklearn.base import clone, BaseEstimator from typing import List -class Partitioner: +class Partitioner(BaseEstimator): """A class for partitionning the feature space, stratified by level sets of predicted probabilities. @@ -327,7 +327,7 @@ def predict(self, X, y_scores): return labels -class GLEstimator: +class GLEstimator(BaseEstimator): """Estimate the grouping loss of a fitted probabilistic classifier. Parameters