Skip to content

psi_param

psi_param

PSIParam (BaseParam)

Source code in federatedml/param/psi_param.py
class PSIParam(BaseParam):

    def __init__(self, max_bin_num=20, need_run=True, dense_missing_val=None,
                 binning_error=consts.DEFAULT_RELATIVE_ERROR):
        super(PSIParam, self).__init__()
        self.max_bin_num = max_bin_num
        self.need_run = need_run
        self.dense_missing_val = dense_missing_val
        self.binning_error = binning_error

    def check(self):
        assert type(self.max_bin_num) == int and self.max_bin_num > 0, 'max bin must be an integer larger than 0'
        assert type(self.need_run) == bool

        if self.dense_missing_val is not None:
            assert type(self.dense_missing_val) == str or type(self.dense_missing_val) == int or \
                   type(self.dense_missing_val) == float, \
                   'missing value type {} not supported'.format(type(self.dense_missing_val))

        self.check_decimal_float(self.binning_error, "psi's param")
__init__(self, max_bin_num=20, need_run=True, dense_missing_val=None, binning_error=0.0001) special
Source code in federatedml/param/psi_param.py
def __init__(self, max_bin_num=20, need_run=True, dense_missing_val=None,
             binning_error=consts.DEFAULT_RELATIVE_ERROR):
    super(PSIParam, self).__init__()
    self.max_bin_num = max_bin_num
    self.need_run = need_run
    self.dense_missing_val = dense_missing_val
    self.binning_error = binning_error
check(self)
Source code in federatedml/param/psi_param.py
def check(self):
    assert type(self.max_bin_num) == int and self.max_bin_num > 0, 'max bin must be an integer larger than 0'
    assert type(self.need_run) == bool

    if self.dense_missing_val is not None:
        assert type(self.dense_missing_val) == str or type(self.dense_missing_val) == int or \
               type(self.dense_missing_val) == float, \
               'missing value type {} not supported'.format(type(self.dense_missing_val))

    self.check_decimal_float(self.binning_error, "psi's param")

Last update: 2021-12-03
Back to top