Exemplo n.º 1
0
 def _prepare(self, data):
     data = get_single_column(data)
     if self.only_if_categorical and not utils.is_categorical(data):
         return None
     levels = self.levels
     if not levels:
         levels = sorted(set(data))
     return levels
Exemplo n.º 2
0
 def _apply(self, data, fitted_feature):
     factors = fitted_feature.prepped_data
     data = get_single_column(data)
     d = DataFrame(index=data.index)
     if self.include_all:
         facts = list(factors)
     else:
         facts = list(factors)[:-1]
     for f in facts:
         d['%s-%s'%(f, data.name)] = data.map(lambda x: int(x == f))
     return d
Exemplo n.º 3
0
 def _prepare(self, data):
     try:
         data = get_single_column(data)
     except ValueError:
         return None
     if self.only_if_categorical and not utils.is_categorical(data):
         return None
     levels = self.levels
     if not levels:
         levels = sorted(set(data))
     return levels
Exemplo n.º 4
0
 def _apply(self, data, fitted_feature):
     factors = fitted_feature.prepped_data
     if factors is None:
         return data
     data = get_single_column(data)
     d = DataFrame(index=data.index)
     if self.include_all:
         facts = list(factors)
     else:
         facts = list(factors)[:-1]
     for f in facts:
         d['%s-%s' % (f, data.name)] = data.map(lambda x: int(x == f))
     return d
Exemplo n.º 5
0
 def _prepare(self, data):
     levels = self.levels
     if not levels:
         levels = sorted(set(get_single_column(data)))
     return levels
Exemplo n.º 6
0
 def _prepare(self, data):
     levels = self.levels
     if not levels:
         levels = set(get_single_column(data))
         levels = zip(levels, range(len(levels)))
     return levels
Exemplo n.º 7
0
def build_target_safe(target, data, prep_index=None, train_index=None):
    y, ff = target.build(data, prep_index, train_index)
    return get_single_column(y), ff
Exemplo n.º 8
0
def apply_target_safe(target, data, fitted_feature):
    y = target.apply(data, fitted_feature)
    return get_single_column(y)
Exemplo n.º 9
0
def build_target_safe(target, data, prep_index=None, train_index=None):
    y, ff = target.build(data, prep_index, train_index)
    return get_single_column(y), ff
Exemplo n.º 10
0
def apply_target_safe(target, data, fitted_feature):
    y = target.apply(data, fitted_feature)
    return get_single_column(y)
Exemplo n.º 11
0
 def _prepare(self, data):
     levels = self.levels
     if not levels:
         levels = set(get_single_column(data))
         levels = zip(levels, range(len(levels)))
     return levels