klasa publiczna LabelUtil
Operacja etykiety utils.
Metody publiczne
static List<ciąg znaków> |
mapValueToLabels(TensorBuffer tensorBuffer, etykiety List<ciąg znaków>, przesunięcie int)
Mapuje tensor wartości int na listę etykiet ciągów znaków.
|
Metody dziedziczone
Metody publiczne
public static List<String> mapValueToLabels (TensorBuffer tensorBuffer, List<Ciąg> etykiety, przesunięcie int)
Mapuje tensor wartości int na listę etykiet ciągów znaków. Używa tablicy ciągów znaków jako słownika. Przykład: jeśli dany tensor to [3, 1, 0], a dane etykiety to ["tło", "jabłko", "banan", "wiśnia", "data"], wynik będzie miał postać ["data", "banan", "jabłko"].
Parametry
tensorBuffer | Tensor z wartościami indeksu. Wartości powinny być nieujemnymi liczbami całkowitymi, a każda wartość x zostanie przekonwertowana na wartość labels[x + offset] . Jeśli tensor jest podany w postaci liczby zmiennoprzecinkowej TensorBuffer , wartości zostaną rzutowane na liczby całkowite. Wszystkie wartości, które są poza zakresem, zostaną zmapowane na pusty ciąg znaków. |
---|---|
etykiety | Lista ciągów tekstowych używanych jako słownik do wyszukiwania. Jako klucz będzie używany indeks elementu tablicy. Aby uzyskać lepszą wydajność, użyj obiektu, który implementuje funkcję RandomAccess, np. ArrayList . |
przesunięcie | Wartość przesunięcia podczas wyszukiwania wartości całkowitych w elemencie labels . |
Zwroty
- zmapowane ciągi. Lista ma długość
TensorBuffer.getFlatSize()
.
Rzuty
IllegalArgumentException | jeśli tensorBuffer lub labels ma wartość null.
|
---|