{ "cells": [ { "cell_type": "markdown", "id": "fff4734f", "metadata": {}, "source": [ "# TensorflowHub\n", "Let's load the TensorflowHub Embedding class." ] }, { "cell_type": "code", "execution_count": 1, "id": "f822104b", "metadata": {}, "outputs": [], "source": [ "from langchain.embeddings import TensorflowHubEmbeddings" ] }, { "cell_type": "code", "execution_count": 5, "id": "bac84e46", "metadata": {}, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "2023-01-30 23:53:01.652176: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX2 FMA\n", "To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.\n", "2023-01-30 23:53:34.362802: I tensorflow/core/platform/cpu_feature_guard.cc:193] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX2 FMA\n", "To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.\n" ] } ], "source": [ "embeddings = TensorflowHubEmbeddings()" ] }, { "cell_type": "code", "execution_count": 6, "id": "4790d770", "metadata": {}, "outputs": [], "source": [ "text = \"This is a test document.\"" ] }, { "cell_type": "code", "execution_count": 7, "id": "f556dcdb", "metadata": {}, "outputs": [], "source": [ "query_result = embeddings.embed_query(text)" ] }, { "cell_type": "code", "execution_count": 6, "id": "76f1b752", "metadata": {}, "outputs": [], "source": [ "doc_results = embeddings.embed_documents([\"foo\"])" ] }, { "cell_type": "code", "execution_count": null, "id": "fff99b21", "metadata": {}, "outputs": [], "source": [ "doc_results" ] }, { "cell_type": "code", "execution_count": null, "id": "aaad49f8", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.9.1" }, "vscode": { "interpreter": { "hash": "7377c2ccc78bc62c2683122d48c8cd1fb85a53850a1b1fc29736ed39852c9885" } } }, "nbformat": 4, "nbformat_minor": 5 }