Omówienie biblioteki centrum TensorFlow

Biblioteka tensorflow_hub umożliwia pobieranie i ponowne wykorzystywanie przeszkolonych modeli w programie TensorFlow przy minimalnej ilości kodu. Głównym sposobem ładowania przeszkolonego modelu jest użycie interfejsu API hub.KerasLayer .

import tensorflow_hub as hub

embed = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)

Uwaga: w tej dokumentacji w przykładach zastosowano uchwyty adresów URL TFhub.dev. Więcej informacji na temat innych prawidłowych typów uchwytów znajdziesz tutaj .

Ustawianie lokalizacji pamięci podręcznej dla pobranych plików.

Domyślnie tensorflow_hub używa ogólnosystemowego katalogu tymczasowego do buforowania pobranych i nieskompresowanych modeli. Zobacz Buforowanie , aby zapoznać się z opcjami użycia innych, prawdopodobnie bardziej trwałych lokalizacji.

Stabilność API

Chociaż mamy nadzieję zapobiec istotnym zmianom, projekt ten jest nadal w fazie aktywnego rozwoju i nie ma jeszcze gwarancji, że będzie miał stabilny interfejs API lub format modelu.

Uczciwość

Podobnie jak w przypadku całego uczenia maszynowego, ważną kwestią jest uczciwość . Wiele wstępnie wyszkolonych modeli jest szkolonych na dużych zestawach danych. Podczas ponownego wykorzystywania dowolnego modelu należy pamiętać, na jakich danych model został przeszkolony (i czy istnieją w nim jakieś błędy systematyczne) oraz jaki może to mieć wpływ na jego użycie.

Bezpieczeństwo

Ponieważ zawierają dowolne wykresy TensorFlow, modele można traktować jako programy. Korzystanie z TensorFlow Securely opisuje konsekwencje bezpieczeństwa wynikające z odwoływania się do modelu z niezaufanego źródła.

Następne kroki