TensorFlow Similarity度量學(xué)習(xí)

TensorFlow Similarity 是一個(gè)用于相似性學(xué)習(xí)的 TensorFLow 庫，也被稱為度量學(xué)習(xí)和對比性學(xué)習(xí)。目前仍處于測試階段。

Tensorflow Similarity 提供了先進(jìn)的度量學(xué)習(xí)算法和所有必要的組件來研究、訓(xùn)練、評估和服務(wù)基于相似性的模型。

通過TensorFlow的相似性，你可以訓(xùn)練并提供模型，在大量的例子語料庫中找到類似的項(xiàng)目（如圖像）。例如，如上圖所示，你可以訓(xùn)練一個(gè)相似性模型，從 Oxford IIIT 寵物數(shù)據(jù)集中尋找和聚類看起來相似的貓和狗的圖像，只需在幾個(gè)類上進(jìn)行訓(xùn)練。要訓(xùn)??練你自己的相似性模型，可參閱此 notebook。

度量學(xué)習(xí)不同于傳統(tǒng)分類，因?yàn)樗哪繕?biāo)不同。該模型學(xué)習(xí)以監(jiān)督或自我監(jiān)督的方式最小化相似示例之間的距離并最大化不同示例之間的距離。無論哪種方式，TensorFlow Similarity 都提供了必要的損失、指標(biāo)、采樣器、可視化器和索引子系統(tǒng)，以使其快速簡便。

目前，TensorFlow Similarity 支持監(jiān)督訓(xùn)練。在未來的版本中，它將支持半監(jiān)督和自監(jiān)督訓(xùn)練。