Kā kodēt novērtējuma raksturlielumu no 1 līdz 5?

Novērtējums no 1 līdz 5 ir ordināla īpašība, kuru var kodēt dažādos veidos atkarībā no uzdevuma un modeļa:

• Veselu skaitļu kodēšana — vienkārši izmantojiet skaitļus 1, 2, 3, 4, 5. Piemērots, ja modelis ņem vērā secību un attālumus starp vērtībām (piemēram, lineārā regresija).

• One-Hot kodēšana — izveidojiet 5 bināras īpašības, katra atbilst vienai vērtējumam. Noderīgi, ja modelim nevajag uztvert secību kā skaitlisku atšķirību (piemēram, lēmumu koki).

• Ordināla kodēšana ar secības saglabāšanu — var izmantot skaitļus, bet ņemot vērā, ka starpība starp 1 un 2 var nebūt vienāda ar starpību starp 4 un 5.

• Target vai vidējās vērtības kodēšana — aizvietojiet vērtējumu ar mērķa mainīgā vidējo vērtību katram vērtējumam, ja uzdevums ir prognozēšana.

Piemērs ar veselu skaitļu kodēšanu Go:

ratings := []int{1, 3, 5, 2, 4}
// Izmantojiet tieši tā, kā ir, modeļa vajadzībām

One-Hot kodēšanai var izveidot 5 elementu masīvu, kur pozīcija atbilst vērtējumam:

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

// Piemērs lietošanai
encoded := oneHotEncode(3) // [0, 0, 1, 0, 0]