Missed TensorFlow World? Check out the recap. Learn more

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

  !pip install "tensorflow_hub==0.4.0"
  !pip install "tf-nightly"

  import tensorflow as tf
  import tensorflow_hub as hub

  tf.enable_eager_execution()

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