TensorFlow Similarity 是一个用于相似性学习的 TensorFLow 库,也被称为度量学习和对比性学习。目前仍处于测试阶段。

Tensorflow Similarity 提供了先进的度量学习算法和所有必要的组件来研究、训练、评估和服务基于相似性的模型。通过 TensorFlow 的相似性,你可以训练并提供模型,在大量的例子语料库中找到类似的项目。度量学习不同于传统分类,因为它的目标不同。该模型学习以监督或自我监督的方式最小化相似示例之间的距离并最大化不同示例之间的距离。无论哪种方式,TensorFlow Similarity 都提供了必要的损失、指标、采样器、可视化器和索引子系统,以使其快速简便。目前,TensorFlow Similarity 支持监督训练。在未来的版本中,它将支持半监督和自监督训练。