1. Lära sig
    2. /
  2. Courses
    3. /
  3. Nhập môn Mô hình tuyến tính với Python
Connected

exercise

Bình phương tối thiểu với `statsmodels`

Một số thư viện Python cung cấp các giao diện trừu tượng tiện lợi để bạn không phải luôn xử lý tường minh mọi chi tiết tối ưu hóa mô hình.

Ví dụ, trong bài tập này, bạn sẽ dùng thư viện statsmodels theo một quy trình làm việc cấp cao, tổng quát hơn để xây dựng mô hình bằng tối ưu hóa bình phương tối thiểu (tối thiểu hóa RSS).

Để bạn bắt đầu nhanh hơn, chúng tôi đã nạp sẵn dữ liệu từ x_data, y_data = load_data() và lưu vào một pandas DataFrame với tên cột x_column và y_column bằng lệnh df = pd.DataFrame(dict(x_column=x_data, y_column=y_data))

Instruktioner

100 XP
  • Tạo mô hình ols() với công thức formula="y_column ~ x_column" và dữ liệu data=df, sau đó .fit() mô hình với dữ liệu.
  • Dùng model_fit.predict() để lấy các giá trị y_model.
  • Sử dụng hàm đã cung cấp plot_data_with_model() để vẽ chồng y_data với y_model.
  • Trích xuất các tham số mô hình a0 và a1 từ model_fit.params.
  • Dùng compute_rss_and_plot_fit() để xác nhận các kết quả này nhất quán với các công thức giải tích đã được triển khai bằng numpy.