Как да кодираме характеристиката на рейтинга от 1 до 5?

Оценка от 1 до 5 е ординална характеристика, която може да бъде кодирана по няколко начина в зависимост от задачата и модела:

• Целочислено кодиране — просто използвайте числата 1, 2, 3, 4, 5. Подходящо, ако моделът взема предвид реда и разстоянията между стойностите (например, линейна регресия).

• One-Hot кодиране — създайте 5 бинарни признака, всеки от които съответства на една стойност на оценката. Полезно, ако моделът не трябва да възприема реда като числова разлика (например, дървета за вземане на решения).

• Ординално кодиране с запазване на реда — може да използвате числа, но като вземете предвид, че разликата между 1 и 2 може да не е равна на разликата между 4 и 5.

• Target или средно кодиране — заменете оценката с средната стойност на целевата променлива за всяка оценка, ако задачата е предсказване.

Пример за целочислено кодиране в Go:

ratings := []int{1, 3, 5, 2, 4}
// Използвайте както е за модела

За One-Hot кодиране може да създадете масив от 5 елемента, където позицията съответства на оценката:

func oneHotEncode(rating int) []int {
    encoded := make([]int, 5)
    if rating >= 1 && rating <= 5 {
        encoded[rating-1] = 1
    }
    return encoded
}

// Пример за използване
encoded := oneHotEncode(3) // [0, 0, 1, 0, 0]