12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- from pipeline.param.base_param import BaseParam
- class PredictParam(BaseParam):
- """
- Define the predict method of HomoLR, HeteroLR, SecureBoosting
- Parameters
- ----------
- threshold: float or int
- The threshold use to separate positive and negative class. Normally, it should be (0,1)
- """
- def __init__(self, threshold=0.5):
- self.threshold = threshold
- def check(self):
- if type(self.threshold).__name__ not in ["float", "int"]:
- raise ValueError("predict param's predict_param {} not supported, should be float or int".format(
- self.threshold))
- return True
|