test_boosted_trees_classifier_numeric.py 文件源码

def setUpClass(self):
        """
        Set up the unit test by loading the dataset and training a model.
        """
        from sklearn.datasets import load_boston

        # Load data and train model
        scikit_data = load_boston()
        self.scikit_data = scikit_data
        self.X = scikit_data.data.astype('f').astype('d') ## scikit-learn downcasts data
        self.target = 1 * (scikit_data['target'] > scikit_data['target'].mean())
        self.feature_names = scikit_data.feature_names
        self.output_name = 'target'