TensorFlow Hub는 재사용 가능한 머신러닝 모듈 라이브러리입니다

  !pip install "tensorflow_hub>=0.6.0"
  !pip install "tensorflow>=2.0.0"

  import tensorflow as tf
  import tensorflow_hub as hub

  module_url = "https://tfhub.dev/google/nnlm-en-dim128/2"
  embed = hub.KerasLayer(module_url)
  embeddings = embed(["A long sentence.", "single-word",
                      "http://example.com"])
  print(embeddings.shape)  #(3,128)
TensorFlow Hub는 머신러닝 모델의 재사용 가능한 부분을 게시, 검색, 소비하기 위한 라이브러리입니다. 모듈은 가중치 및 애셋과 함께 자체 포함된 TensorFlow 그래프 조각인데, 전이 학습으로 알려진 프로세스의 다양한 작업에서 재사용할 수 있습니다. 전이 학습을 통해 다음과 같은 작업이 가능합니다.
  • 소규모 데이터세트를 사용한 모델 학습
  • 일반화 개선
  • 학습 속도 개선