Tương quan

Một đặc điểm thay đổi cùng với mục tiêu sẽ giúp mô hình dự đoán mục tiêu đó. Hệ số tương quan r đo mức độ hai cột cùng biến thiên theo một đường thẳng, từ -1 đến 1:

Sáu tuần quảng cáo và doanh số của một cửa hàng: np.corrcoef trả về một bảng 2 nhân 2, mỗi mảng được so sánh với mảng còn lại, trong đó r nằm ở [0, 1]. Phương thức .corr của một cột trả về giá trị này trực tiếp:

ads = np.array([100, 200, 150, 300, 250, 50])
sales = np.array([430, 520, 560, 760, 610, 410])
print(np.round(np.corrcoef(ads, sales), 2))
shop = pd.DataFrame({"ads": ads, "sales": sales})
print(np.round(shop["ads"].corr(shop["sales"]), 2))

Kết quả in ra:

[[1.   0.94]
 [0.94 1.  ]]
0.94

Ở mức 0.94, những tuần có nhiều quảng cáo hơn rõ ràng bán được nhiều hàng hơn.