Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, values, columns=None, **kwargs):
self._values = values
self._values_str = _list_str(self._values)
self._columns_str = _list_str(columns)
if columns is None:
self._columns = None
apply_msg = ValDrop._DEF_VALDROP_APPLY_MSG.format(
self._values_str)
else:
self._columns = _interpret_columns_param(columns)
apply_msg = ValDrop._DEF_VALDROP_APPLY_MSG.format(
"{} in {}".format(
self._values_str, self._columns_str))
super_kwargs = {
'exmsg': ValDrop._DEF_VALDROP_EXC_MSG.format(self._columns_str),
'appmsg': apply_msg,
'desc': self._default_desc()
}
super_kwargs.update(**kwargs)
def __init__(self, columns, threshold, drop=True, **kwargs):
self._columns = _interpret_columns_param(columns)
self._threshold = threshold
self._drop = drop
self._rare_removers = {}
col_str = _list_str(self._columns)
super_kwargs = {
'exmsg': DropRareTokens._DEF_RARE_EXC_MSG.format(col_str),
'appmsg': "Dropping rare tokens from {}...".format(col_str),
'desc': "Drop rare tokens from {}".format(col_str)
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(
self, columns=None, exclude_columns=None, drop=True, **kwargs
):
if columns is None:
self._columns = None
else:
self._columns = _interpret_columns_param(columns)
if exclude_columns is None:
self._exclude_columns = []
else:
self._exclude_columns = _interpret_columns_param(exclude_columns)
self._drop = drop
self.encoders = {}
col_str = _list_str(self._columns)
super_kwargs = {
"exmsg": Encode._DEF_ENCODE_EXC_MSG.format(col_str),
"appmsg": Encode._DEF_ENCODE_APP_MSG.format(col_str),
"desc": "Encode {}".format(col_str or "all categorical columns"),
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
self._columns = None
if columns:
self._columns = _interpret_columns_param(columns)
if reduce not in RowDrop._REDUCERS.keys():
raise ValueError((
"{} is an unsupported argument for the 'reduce' parameter of "
"the RowDrop constructor!").format(reduce))
self._cond_is_dict = isinstance(conditions, dict)
self._columns_str = ""
if self._cond_is_dict:
valid = all([callable(cond) for cond in conditions.values()])
if not valid:
raise ValueError(
"Condition dicts given to RowDrop must map to callables!")
self._columns = list(conditions.keys())
self._columns_str = _list_str(self._columns)
else:
valid = all([callable(cond) for cond in conditions])
if not valid:
raise ValueError(
"RowDrop condition lists can contain only callables!")
self._row_cond = self._row_condition_builder(conditions, reduce)
super_kwargs = {
'exmsg': RowDrop._DEF_ROWDROP_EXC_MSG.format(self._columns_str),
'appmsg': RowDrop._DEF_ROWDROP_APPLY_MSG.format(self._columns_str),
'desc': self._default_desc()
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(self, values, columns=None, **kwargs):
self._values = values
self._values_str = _list_str(self._values)
self._columns_str = _list_str(columns)
if columns is None:
self._columns = None
apply_msg = ValKeep._DEF_VALKEEP_APPLY_MSG.format(
self._values_str)
else:
self._columns = _interpret_columns_param(columns)
apply_msg = ValKeep._DEF_VALKEEP_APPLY_MSG.format(
"{} in {}".format(
self._values_str, self._columns_str))
super_kwargs = {
'exmsg': ValKeep._DEF_VALKEEP_EXC_MSG.format(self._columns_str),
'appmsg': apply_msg,
'desc': self._default_desc()
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(self, rename_map, **kwargs):
self._rename_map = rename_map
columns_str = _list_str(list(rename_map.keys()))
suffix = 's' if len(rename_map) > 1 else ''
super_kwargs = {
'exmsg': ColRename._DEF_COLDRENAME_EXC_MSG.format(columns_str),
'appmsg': ColRename._DEF_COLDRENAME_APP_MSG.format(
suffix, columns_str),
'desc': "Rename column{} with {}".format(suffix, self._rename_map)
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(self, stemmer_name, columns, drop=True, **kwargs):
self.stemmer_name = stemmer_name
self.stemmer = SnowballStem.__safe_stemmer_by_name(stemmer_name)
self.list_stemmer = SnowballStem._TokenListStemmer(self.stemmer)
self._columns = _interpret_columns_param(columns)
col_str = _list_str(self._columns)
super_kwargs = {
'columns': columns,
'value_map': self.list_stemmer,
'drop': drop,
'suffix': '_stem',
'exmsg': SnowballStem._DEF_STEM_EXC_MSG.format(col_str),
'appmsg': SnowballStem._DEF_STEM_APP_MSG.format(col_str),
'desc': "Stem tokens in {}".format(col_str),
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(self, bin_map, drop=True, **kwargs):
self._bin_map = bin_map
self._drop = drop
columns_str = _list_str(list(bin_map.keys()))
super_kwargs = {
"exmsg": Bin._DEF_BIN_EXC_MSG.format(columns_str),
"appmsg": Bin._DEF_BIN_APP_MSG.format(
"s" if len(bin_map) > 1 else "", columns_str
),
"desc": self._default_desc(),
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)
def __init__(
self,
scaler,
exclude_columns=None,
exclude_object_columns=True,
**kwargs
):
self.scaler = scaler
if exclude_columns is None:
self._exclude_columns = []
desc_suffix = "."
else:
self._exclude_columns = _interpret_columns_param(exclude_columns)
col_str = _list_str(self._exclude_columns)
desc_suffix = " except columns {}.".format(col_str)
self._exclude_obj_cols = exclude_object_columns
super_kwargs = {
"exmsg": Scale._DEF_SCALE_EXC_MSG,
"appmsg": Scale._DEF_SCALE_APP_MSG,
"desc": Scale._DESC_PREFIX + desc_suffix,
}
self._kwargs = kwargs
valid_super_kwargs = super()._init_kwargs()
for key in kwargs:
if key in valid_super_kwargs:
super_kwargs[key] = kwargs[key]
super().__init__(**super_kwargs)
def __init__(self, values, columns=None, **kwargs):
self._values = values
self._values_str = _list_str(self._values)
self._columns_str = _list_str(columns)
if columns is None:
self._columns = None
apply_msg = ValDrop._DEF_VALDROP_APPLY_MSG.format(
self._values_str)
else:
self._columns = _interpret_columns_param(columns)
apply_msg = ValDrop._DEF_VALDROP_APPLY_MSG.format(
"{} in {}".format(
self._values_str, self._columns_str))
super_kwargs = {
'exmsg': ValDrop._DEF_VALDROP_EXC_MSG.format(self._columns_str),
'appmsg': apply_msg,
'desc': self._default_desc()
}
super_kwargs.update(**kwargs)
super().__init__(**super_kwargs)