Kom igångKom igång gratis

Visualisera en linjär regressionsmodell

Nu när du har byggt din linjära regressionsmodell och tränat den på alla tillgängliga observationer kan du visualisera hur väl modellen passar data. Det gör det lättare att tolka sambandet mellan reklamutgifter för radio och värden för sales.

Variablerna X, en array med radio-värden, y, en array med sales-värden, och predictions, en array med modellens förutsagda värden för y givet X, har alla förhandsladdats åt dig från föregående övning.

Den här övningen är en del av kursen

Övervakad inlärning med scikit-learn

Visa kurs

Övningsinstruktioner

  • Importera matplotlib.pyplot som plt.
  • Skapa ett punktdiagram som visar y mot X, där observationerna är blå.
  • Rita ett rött linjediagram som visar förutsägelserna mot X.
  • Visa diagrammet.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Import matplotlib.pyplot
import ____.____ as ____

# Create scatter plot
plt.scatter(____, ____, color="____")

# Create line plot
plt.plot(____, ____, color="____")
plt.xlabel("Radio Expenditure ($)")
plt.ylabel("Sales ($)")

# Display the plot
plt.____()
Redigera och kör kod