۶ کیو· متوسطمتوسطPYTHON۴ ثانیه · ۲۵۶ مگابایت

مدل با جمله‌ی تعامل

تابع fit_with_interaction(x1, x2, y) مدل

y = \beta_0 + \beta_1 x_1 + \beta_2 x_2 + \beta_3 \, x_1 x_2 + \varepsilon

را با کمترین مربعات برازش می‌دهد و [b0, b1, b2, b3] را برمی‌گرداند.

ماتریس طرح را خودتان بسازید: ستون یک، $x_1$ ، $x_2$ و حاصل‌ضرب عضوبه‌عضو $x_1 x_2$ . برای حل، np.linalg.lstsq(A, y, rcond=None)[0] یا معادلات نرمال هر دو درست‌اند.

نمونه‌ها

ورودی نمونه ۱

[[0, 1, 0, 1, 2], [0, 0, 1, 1, 2], [1, 3, 2, 7, 15]]

خروجی نمونه ۱

[0.8462, 2.4615, 1.4615, 1.6154]

ورودی نمونه ۲

[[1, 2, 3, 1, 2, 3], [0, 0, 0, 1, 1, 1], [2.0, 2.9, 4.1, 3.0, 5.1, 6.9]]

خروجی نمونه ۲

[0.9, 1.05, 0.2, 0.9]

برای نوشتن پاسخ و دریافت امتیاز، ابتدا وارد حساب شوید.

ورود