질문이있다? TensorFlow 포럼 방문 포럼 에서 커뮤니티와 연결

TensorFlow 허브

TensorFlow Hub는 재사용 가능한 기계 학습을위한 개방형 저장소 및 라이브러리입니다. tfhub.dev 저장소는 텍스트 임베딩, 이미지 분류 모델, TF.js / TFLite 모델 등 많은 사전 학습 된 모델을 제공합니다. 저장소는 커뮤니티 기여자 에게 열려 있습니다 .

tensorflow_hub 라이브러리를 사용하면 최소한의 코드로 TensorFlow 프로그램에서 다운로드하고 재사용 할 수 있습니다.

import tensorflow_hub as hub

model = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = model(["The rain in Spain.", "falls",
                    "mainly", "In the plain!"])

print(embeddings.shape)  #(4,128)

다음 단계