Hogyan kódoljuk az 1-től 5-ig terjedő értékelési jellemzőt?

Az 1-től 5-ig terjedő értékelés egy ordinális jellemző, amelyet különböző módon lehet kódolni a feladattól és a modelltől függően:

• Egész számkódolás — egyszerűen használja az 1, 2, 3, 4, 5 számokat. Akkor megfelelő, ha a modell figyelembe veszi a sorrendet és a távolságokat az értékek között (pl. lineáris regresszió).

• One-Hot kódolás — hozzon létre 5 bináris jellemzőt, amelyek mindegyike megfelel egy értékelési értéknek. Hasznos, ha a modell nem kell, hogy észlelje a sorrendet numerikus különbségként (pl. döntési fák).

• Sorrendi kódolás a sorrend megőrzésével — használhat számokat, de figyelembe véve, hogy a 1 és 2 közötti különbség nem feltétlenül egyezik meg a 4 és 5 közötti különbséggel.

• Target vagy átlagos kódolás — helyettesítse az értékelést a célváltozó átlagértékével minden értékelés esetén, ha a feladat predikció.

Go nyelvben például így néz ki az egész számkódolás:

ratings := []int{1, 3, 5, 2, 4}
// Használja pontosan így a modellhez

One-Hot kódoláshoz létrehozhat egy 5 elemű tömböt, ahol a pozíció megfelel az értékelésnek:

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

// Példa a használatra
encoded := oneHotEncode(3) // [0, 0, 1, 0, 0]