۶ کیو· متوسطمتوسطPYTHON۴ ثانیه · ۲۵۶ مگابایت
انتخاب بهترین زیرمجموعه
تابع best_subset(X, y, k) از میان همه‌ی زیرمجموعه‌های تایی ستون‌های ، آن را پیدا می‌کند که رگرسیون کمترین مربعاتِ روی آن ستون‌ها — با عرض از مبدأ — کمترین RSS را دارد.
خروجی آرایه‌ای به طول است: شماره‌ی ستون‌های بهترین زیرمجموعه به ترتیب صعودی، و بعد RSS همان مدل:
text
[j1, j2, ..., jk, rss]
itertools.combinations(range(p), k) همه‌ی زیرمجموعه‌ها را می‌دهد. در آزمون‌ها بهترین زیرمجموعه یکتاست.
نمونه‌ها
ورودی نمونه ۱
[[[3.53, 4.67, 4.04, 6.2], [5.08, 4.42, 3.44, 4.49], [5.02, 4.45, 7.59, 7.01], [-0.42, 1.22, 4.65, 4.16], [5.43, 5.43, 9.24, 2.78], [4.24, 9.09, 6.29, 6.33], [3.97, 1.7, 5.33, 5.22], [2.55, 3.63, 4.86, 3.11], [4.8, 5.19, 5.07, 3.99], [6.19, 6.78, 5.64, 3.36], [6.46, 4.0, 6.76, 2.86], [6.83, 4.96, 2.5, 4.37]], [4.05, 8.27, 0.67, -5.92, 0.2, 1.58, 3.18, 1.57, 3.35, 7.17, 7.0, 12.61], 2]
خروجی نمونه ۱
[0.0, 2.0, 6.6473]
ورودی نمونه ۲
[[[4.8, 6.09, 3.79, 5.25, 3.22], [6.68, 5.38, 5.66, 5.82, 2.98], [6.57, 9.11, 1.72, 1.54, 1.99], [6.68, 5.26, 7.16, 6.44, 5.42], [5.57, 4.66, 6.74, 2.74, 4.16], [5.49, 8.6, 3.47, 2.84, 3.87], [6.94, 4.53, 7.65, 1.25, 7.26], [7.07, 2.16, 5.31, 7.43, 5.18], [7.0, 9.75, 5.55, 4.44, 3.46], [6.3, 4.61, 4.64, 4.79, 6.3], [2.87, 1.94, 0.13, 7.4, 5.15], [8.02, 4.98, 3.52, 5.96, 4.85], [2.49, 3.23, 8.53, 5.71, 5.83], [4.45, 3.62, 6.78, 4.79, 3.48], [4.73, 3.19, 5.38, 7.26, 3.33]], [8.87, 6.54, 12.46, 4.31, 5.09, 10.26, 2.28, 0.89, 11.39, 2.84, -0.05, 4.89, 2.53, 5.29, 4.68], 1]
خروجی نمونه ۲
[1.0, 26.6306]
برای نوشتن پاسخ و دریافت امتیاز، ابتدا وارد حساب شوید.
ورود