Breaking News: Grepper is joining You.com. Read the official announcement!

yellow pbrick plot auc curve

Gleaming Goat answered on February 28, 2023 Popularity 1/10 Helpfulness 1/10

answer yellow pbrick plot auc curve

yellow pbrick plot auc curve

Comment

from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split

from yellowbrick.classifier import ROCAUC
from yellowbrick.datasets import load_spam

# Load the classification dataset
X, y = load_spam()

# Create the training and test data
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=42)

# Instantiate the visualizer with the classification model
model = LogisticRegression(multi_class="auto", solver="liblinear")
visualizer = ROCAUC(model, classes=["not_spam", "is_spam"])

visualizer.fit(X_train, y_train)        # Fit the training data to the visualizer
visualizer.score(X_test, y_test)        # Evaluate the model on the test data
visualizer.show()                       # Finalize and show the figure

Popularity 1/10 Helpfulness 1/10 Language python

Source: www.scikit-yb.org

Tags: auc curve plot python

Link to this answer
Share Copy Link