ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-07 18:08:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Make classifiers and regressors clonable

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-01 07:41:59 -05:00
parent ac29b5213f
commit 820a6256c2
7 changed files with 62 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
miplearn/classifiers/threshold.py
View File

@@ -49,6 +49,13 @@ class Threshold(ABC):
"""
pass
@abstractmethod
def clone(self) -> "Threshold":
"""
Returns an unfitted copy of this threshold with the same hyperparameters.
"""
pass
class MinProbabilityThreshold(Threshold):
"""
@@ -65,6 +72,9 @@ class MinProbabilityThreshold(Threshold):
def predict(self, x_test: np.ndarray) -> List[float]:
return self.min_probability
def clone(self) -> "MinProbabilityThreshold":
return MinProbabilityThreshold(self.min_probability)
class MinPrecisionThreshold(Threshold):
"""
@@ -111,3 +121,8 @@ class MinPrecisionThreshold(Threshold):
if precision[k] >= min_precision:
return thresholds[k]
return float("inf")
def clone(self) -> "MinPrecisionThreshold":
return MinPrecisionThreshold(
min_precision=self.min_precision,
)