| 1234567891011121314151617181920212223242526272829303132333435363738 |
- #!/usr/bin/env python
- # -*- coding: utf-8 -*-
- #
- # added by jsweng
- # param class for OHE with alignment
- #
- from federatedml.param.base_param import BaseParam
- class HomoOneHotParam(BaseParam):
- """
- Parameters
- ----------
- transform_col_indexes: list or int, default: -1
- Specify which columns need to calculated. -1 represent for all columns.
- need_run: bool, default True
- Indicate if this module needed to be run
- need_alignment: bool, default True
- Indicated whether alignment of features is turned on
- """
- def __init__(self, transform_col_indexes=-1, transform_col_names=None, need_run=True, need_alignment=True):
- super(HomoOneHotParam, self).__init__()
- self.transform_col_indexes = transform_col_indexes
- self.transform_col_names = transform_col_names
- self.need_run = need_run
- self.need_alignment = need_alignment
- def check(self):
- descr = "One-hot encoder with alignment param's"
- self.check_defined_type(self.transform_col_indexes, descr, ['list', 'int'])
- self.check_boolean(self.need_run, descr)
- self.check_boolean(self.need_alignment, descr)
- self.transform_col_names = [] if self.transform_col_names is None else self.transform_col_names
- return True
|
