2f2e5a2534e105f07e2f0db40f67e4e4eb8cff92,yellowbrick/classifier/classification_report.py,,classification_report,#,290

Before Change

    )

    // Create the train and test splits
    X_train, X_test, y_train, y_test = train_test_split(
        X, y, test_size=test_size, random_state=random_state
    )

    // Fit and transform the visualizer (calls draw)
    visualizer.fit(X_train, y_train, **kwargs)
    visualizer.score(X_test, y_test)

After Change

    visualizer.fit(X_train, y_train)

    // Score the visualizer
    if X_test is not None and y_test is not None:
        visualizer.score(X_test, y_test)
    elif X_test is not None or y_test is not None:
        raise YellowbrickValueError(
            "both X_test and y_test are required if one is specified"
        )
    else:
        visualizer.score(X_train, y_train)
    
    // Draw the final visualization
    if show:
        visualizer.show()
    else:
        visualizer.finalize()

    // Return the visualizer
    return visualizer

Instances