klasa publiczna LabelUtil
Narzędzia obsługi etykiet.
Metody publiczne
lista statyczna < ciąg > | mapValueToLabels ( TensorBuffer tensorBuffer, List < String > etykiety, przesunięcie int) Mapuje tensor wartości int na listę etykiet łańcuchowych. |
Metody dziedziczone
Metody publiczne
public static List < String > mapValueToLabels ( TensorBuffer tensorBuffer, List < String > labels, int offset)
Mapuje tensor wartości int na listę etykiet łańcuchowych. Jako słownik przyjmuje tablicę ciągów znaków. Przykład: jeśli podany tensor to [3, 1, 0], a etykieta to ["tło", "jabłko", "banan", "wiśnia", "data"], wynikiem będzie ["data", „banan”, „jabłko”].
Parametry
bufor tensora | Tensor z wartościami indeksów. Wartości powinny być nieujemnymi liczbami całkowitymi, a każda wartość x zostanie przekonwertowana na labels[x + offset] . Jeśli tensor jest podany jako float TensorBuffer , wartości zostaną rzutowane na liczby całkowite. Wszystkie wartości, które są poza granicami, zostaną odwzorowane na pusty ciąg. |
---|---|
etykiety | Lista ciągów znaków używana jako słownik do wyszukiwania. Jako klucz zostanie użyty indeks elementu tablicy. Aby uzyskać lepszą wydajność, użyj obiektu implementującego RandomAccess, takiego jak ArrayList . |
zrównoważyć | Wartość przesunięcia podczas wyszukiwania wartości int w labels . |
Zwroty
- mapowane ciągi. Długość listy to
TensorBuffer.getFlatSize()
.
Rzuca
Wyjątek IllegalArgument | jeśli tensorBuffer lub labels mają wartość null. |
---|