test_learning_curve.py 文件源码

def test_learning_curve_comprehensive(self):
        """
        Test learning curve with all parameters with visual unit test.
        """

        try:
            visualizer = LearningCurveVisualizer(LinearSVC(random_state=0), train_sizes=np.linspace(.1, 1.0, 5), 
                cv=ShuffleSplit(n_splits=100, test_size=0.2, random_state=0), 
                n_jobs=4)
            visualizer.fit(X, y)
            visualizer.poof()
        except Exception as e:
            self.fail("error during learning curve: {}".format(e))

        self.assert_images_similar(visualizer)