a509d63e8b05f66150afcfa9216b2cbd9b01ce9c,yellowbrick/classifier/rocauc.py,,roc_auc,#,453

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)
    visualizer.finalize()

    // Return the visualizer
    return visualizer

After Change

    visualizer.fit(X_train, y_train, **kwargs)

    //Scores the visualizer with X_test and y_test if provided, X_train, y_train if not provided
    if X_test is not None and y_test is not None:
        visualizer.score(X_test, y_test)
    else:
        visualizer.score(X_train,  y_train)
    
    if show:
        visualizer.show()
    else:
        visualizer.finalize()

    // Return the visualizer
    return visualizer

Instances