Skapa särdrag
I det här kapitlet arbetar du med en datamängd som heter sales_df. Den innehåller information om reklamkampanjers utgifter för olika medietyper samt antalet dollar som genererats i försäljning för respektive kampanj. Datamängden har förinsatts åt dig. Här är de två första raderna:
tv radio social_media sales
1 13000.0 9237.76 2409.57 46677.90
2 41000.0 15886.45 2913.41 150177.83
Du använder reklamutgifterna som särdrag för att förutsäga försäljningsvärden, och börjar med kolumnen "radio". Innan du kan göra några förutsägelser behöver du skapa särdragsmatrisen och målvektorn, och omforma dem till det format som scikit-learn kräver.
Den här övningen är en del av kursen
Övervakad inlärning med scikit-learn
Övningsinstruktioner
- Skapa
X, en array med värdena från kolumnen"radio"i DataFrame:nsales_df. - Skapa
y, en array med värdena från kolumnen"sales"i DataFrame:nsales_df. - Omforma
Xtill en tvådimensionell NumPy-array. - Skriv ut formen på
Xochy.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
import numpy as np
# Create X from the radio column's values
X = ____
# Create y from the sales column's values
y = ____
# Reshape X
X = ____
# Check the shape of the features and targets
print(____)