test_learning_curve.py 文件源码

def test_learning_curve_model_trainsize_cv_only(self):
        """
        Test learning curve with inputting model, training size, and cv only.
        """

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