Vizualizarea unui model de regresie liniară
Acum că ai construit modelul de regresie liniară și l-ai antrenat folosind toate observațiile disponibile, poți vizualiza cât de bine se potrivește modelul cu datele. Astfel, poți interpreta relația dintre cheltuielile de publicitate radio și valorile sales.
Variabilele X, un array cu valorile radio, y, un array cu valorile sales, și predictions, un array cu valorile prezise de model pentru y dat X, au fost preîncărcate pentru tine din exercițiul anterior.
Acest exercițiu face parte din cursul
Învățare supravegheată cu scikit-learn
Instrucțiuni pentru exercițiu
- Importă
matplotlib.pyplotcaplt. - Creează un grafic scatter care să vizualizeze
yfață deX, cu observațiile reprezentate în albastru. - Trasează un grafic linie roșie care să afișeze predicțiile față de
X. - Afișează graficul.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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.____()