RegressionDecisionTree.py 文件源码

def __init__(self, isTrain):
        super(RegressionDecisionTree, self).__init__(isTrain)
        # data preprocessing
        #self.dataPreprocessing()

        # Create linear regression object
        self.model = DecisionTreeRegressor(max_depth=7, max_features=None)