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

رگرسیون چندگانه با معادلات نرمال

تابع fit_multiple(X, y) ماتریس ویژگی‌ها XX (شکل n×pn \times p، بدون ستون یک) و بردار پاسخ yy را می‌گیرد و ضرایب کمترین مربعات را به‌صورت [b0, b1, ..., bp] برمی‌گرداند — اول عرض از مبدأ.

یک ستون یک به اول XX اضافه کنید و معادلات نرمال را حل کنید:

β^=(X⊤X)−1X⊤y\hat\beta = (X^\top X)^{-1} X^\top y

برای حل، np.linalg.solve(A, b) را به محاسبه‌ی صریح وارون ترجیح دهید. در آزمون‌ها X⊤XX^\top X همیشه وارون‌پذیر است.

نمونه‌ها

ورودی نمونه ۱
[[[1, 0], [0, 1], [1, 1], [2, 1]], [3, 4, 6, 8]]
خروجی نمونه ۱
[1.0, 2.0, 3.0]
ورودی نمونه ۲
[[[1], [2], [3], [4], [5]], [3, 5, 4, 7, 8]]
خروجی نمونه ۲
[1.8, 1.2]

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

ورود