1. Learn
    2. /
  2. Cursuri
    3. /
  3. Învățare supravegheată cu scikit-learn
Connected

exercițiu

Antrenează și prezice pentru regresie

Acum că ai văzut cum funcționează regresia liniară, sarcina ta este să creezi un model de regresie liniară multiplă folosind toate caracteristicile din setul de date sales_df, care a fost preîncărcat pentru tine. Ca reminder, iată primele două rânduri:

     tv        radio      social_media    sales
1    13000.0   9237.76    2409.57         46677.90
2    41000.0   15886.45   2913.41         150177.83

Vei folosi apoi acest model pentru a prezice vânzările pe baza valorilor caracteristicilor de test.

LinearRegression și train_test_split au fost preîncărcate pentru tine din modulele respective.

Instrucțiuni

100 XP
  • Creează X, un array care conține valorile tuturor caracteristicilor din sales_df, și y, care conține toate valorile din coloana "sales".
  • Instanțiază un model de regresie liniară.
  • Antrenează modelul pe datele de antrenament.
  • Creează y_pred, generând predicții pentru sales folosind caracteristicile de test.