۶ کیو· متوسطمتوسطPYTHON۴ ثانیه · ۲۵۶ مگابایت
کدگذاری یک‌داغ
تابع one_hot(train, test) یک ستون دسته‌ای را کدگذاری یک‌داغ (one-hot) می‌کند:
- دسته‌ها را فقط از
trainبگیرید: مقدارهای یکتای آن، مرتب‌شده (sorted(set(train))). - برای هر مقدار
testیک ردیف بسازید با یک ستون برای هر دسته: یک در ستون دسته‌ی خودش و صفر در بقیه. - مقداری از
testکه درtrainنبوده، ردیفی تمام صفر می‌گیرد — همانOneHotEncoder(handle_unknown="ignore").
خروجی ماتریس صفر و یک به شکل (تعداد test، تعداد دسته‌ها) است. ورودی‌ها فهرست رشته‌اند.
نمونه‌ها
ورودی نمونه ۱
[["red", "green", "blue", "green"], ["green", "blue", "red"]]
خروجی نمونه ۱
[[0.0, 1.0, 0.0], [1.0, 0.0, 0.0], [0.0, 0.0, 1.0]]
ورودی نمونه ۲
[["تهران", "شیراز", "تبریز", "شیراز", "تهران"], ["تبریز", "مشهد", "تهران"]]
خروجی نمونه ۲
[[1.0, 0.0, 0.0], [0.0, 0.0, 0.0], [0.0, 1.0, 0.0]]
برای نوشتن پاسخ و دریافت امتیاز، ابتدا وارد حساب شوید.
ورود