xtress_bin.py 文件源码

def __init__(self, train=None, test=None, test_DF=None, name='Apache',
               verbose=True, smoteit=False, majority=False, bin=False):
    self.train, self.test = train, test
    self.name = name
    try:
      self.train_DF = createTbl(train, _smote=smoteit, isBin=False)
    except:
      set_trace()
    if not test_DF:
      self.test_DF = createTbl(test, isBin=False)
    else:
      self.test_DF = test_DF
    self.ignore = [self.train_DF.headers[i].name[1:]
                   for i in avoid(name=name)]
    self.verbose, self.smoteit = verbose, smoteit
    self.mod, self.keys = [], self.getKey()
    self.majority = majority
    t = discreteNums(
        createTbl(train, _smote=smoteit, isBin=bin),
        map(
            lambda x: x.cells,
            self.train_DF._rows))
    self.myTree = tdiv(t)
#     set_trace()