examples
3751 строка · 137.9 Кб
1{
2"cells": [
3{
4"attachments": {},
5"cell_type": "markdown",
6"metadata": {
7"colab_type": "text",
8"id": "view-in-github"
9},
10"source": [
11"[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/pinecone-io/examples/blob/master/integrations/cohere/semantic_search_trec.ipynb) [![Open nbviewer](https://raw.githubusercontent.com/pinecone-io/examples/master/assets/nbviewer-shield.svg)](https://nbviewer.org/github/pinecone-io/examples/blob/master/integrations/cohere/semantic_search_trec.ipynb)"
12]
13},
14{
15"cell_type": "markdown",
16"metadata": {
17"id": "UnbNRZTDNA1M"
18},
19"source": [
20"# Semantic Search with Cohere and Pinecone\n",
21"\n",
22"In this notebook we will demonstrate how to perform semantic search for identifying similar or duplicate questions using Cohere and Pinecone.\n",
23"\n",
24"![Steps in semantic search process](https://raw.githubusercontent.com/pinecone-io/examples/master/integrations/cohere/assets/index_query_pinecone_cohere.png)"
25]
26},
27{
28"cell_type": "markdown",
29"metadata": {
30"id": "0974WnM7NA1O"
31},
32"source": [
33"## Setup"
34]
35},
36{
37"cell_type": "markdown",
38"metadata": {
39"id": "uUqE_6bcNA1O"
40},
41"source": [
42"We first need to setup our environment and retrieve API keys for Cohere and Pinecone. Let's start with our environment, we need HuggingFace *Datasets* for our data, and the Cohere and Pinecone clients:"
43]
44},
45{
46"cell_type": "code",
47"execution_count": null,
48"metadata": {
49"colab": {
50"base_uri": "https://localhost:8080/"
51},
52"id": "Mhmhz2D3NA1P",
53"outputId": "0d9c6c92-19b0-4538-9be8-a0304b4286b6"
54},
55"outputs": [
56{
57"name": "stdout",
58"output_type": "stream",
59"text": [
60"Collecting cohere\n",
61" Downloading cohere-1.3.2.tar.gz (8.0 kB)\n",
62"Collecting pinecone-client\n",
63" Downloading pinecone_client-2.0.8-py3-none-any.whl (149 kB)\n",
64"\u001b[K |████████████████████████████████| 149 kB 7.9 MB/s \n",
65"\u001b[?25hCollecting datasets\n",
66" Downloading datasets-2.0.0-py3-none-any.whl (325 kB)\n",
67"\u001b[K |████████████████████████████████| 325 kB 47.9 MB/s \n",
68"\u001b[?25hRequirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from cohere) (2.23.0)\n",
69"Requirement already satisfied: python-dateutil>=2.5.3 in /usr/local/lib/python3.7/dist-packages (from pinecone-client) (2.8.2)\n",
70"Collecting dnspython>=2.0.0\n",
71" Downloading dnspython-2.2.1-py3-none-any.whl (269 kB)\n",
72"\u001b[K |████████████████████████████████| 269 kB 48.1 MB/s \n",
73"\u001b[?25hRequirement already satisfied: typing-extensions>=3.7.4 in /usr/local/lib/python3.7/dist-packages (from pinecone-client) (3.10.0.2)\n",
74"Requirement already satisfied: urllib3>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from pinecone-client) (1.24.3)\n",
75"Collecting pyyaml>=5.4\n",
76" Downloading PyYAML-6.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (596 kB)\n",
77"\u001b[K |████████████████████████████████| 596 kB 48.4 MB/s \n",
78"\u001b[?25hCollecting loguru>=0.5.0\n",
79" Downloading loguru-0.6.0-py3-none-any.whl (58 kB)\n",
80"\u001b[K |████████████████████████████████| 58 kB 5.3 MB/s \n",
81"\u001b[?25hRequirement already satisfied: six>=1.5 in /usr/local/lib/python3.7/dist-packages (from python-dateutil>=2.5.3->pinecone-client) (1.15.0)\n",
82"Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->cohere) (3.0.4)\n",
83"Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->cohere) (2021.10.8)\n",
84"Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->cohere) (2.10)\n",
85"Requirement already satisfied: multiprocess in /usr/local/lib/python3.7/dist-packages (from datasets) (0.70.12.2)\n",
86"Collecting responses<0.19\n",
87" Downloading responses-0.18.0-py3-none-any.whl (38 kB)\n",
88"Requirement already satisfied: packaging in /usr/local/lib/python3.7/dist-packages (from datasets) (21.3)\n",
89"Collecting xxhash\n",
90" Downloading xxhash-3.0.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (212 kB)\n",
91"\u001b[K |████████████████████████████████| 212 kB 48.2 MB/s \n",
92"\u001b[?25hRequirement already satisfied: numpy>=1.17 in /usr/local/lib/python3.7/dist-packages (from datasets) (1.21.5)\n",
93"Requirement already satisfied: pandas in /usr/local/lib/python3.7/dist-packages (from datasets) (1.3.5)\n",
94"Collecting aiohttp\n",
95" Downloading aiohttp-3.8.1-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (1.1 MB)\n",
96"\u001b[K |████████████████████████████████| 1.1 MB 39.1 MB/s \n",
97"\u001b[?25hRequirement already satisfied: tqdm>=4.62.1 in /usr/local/lib/python3.7/dist-packages (from datasets) (4.63.0)\n",
98"Collecting fsspec[http]>=2021.05.0\n",
99" Downloading fsspec-2022.2.0-py3-none-any.whl (134 kB)\n",
100"\u001b[K |████████████████████████████████| 134 kB 47.5 MB/s \n",
101"\u001b[?25hRequirement already satisfied: importlib-metadata in /usr/local/lib/python3.7/dist-packages (from datasets) (4.11.3)\n",
102"Collecting huggingface-hub<1.0.0,>=0.1.0\n",
103" Downloading huggingface_hub-0.4.0-py3-none-any.whl (67 kB)\n",
104"\u001b[K |████████████████████████████████| 67 kB 2.3 MB/s \n",
105"\u001b[?25hRequirement already satisfied: dill in /usr/local/lib/python3.7/dist-packages (from datasets) (0.3.4)\n",
106"Requirement already satisfied: pyarrow>=5.0.0 in /usr/local/lib/python3.7/dist-packages (from datasets) (6.0.1)\n",
107"Requirement already satisfied: filelock in /usr/local/lib/python3.7/dist-packages (from huggingface-hub<1.0.0,>=0.1.0->datasets) (3.6.0)\n",
108"Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.7/dist-packages (from packaging->datasets) (3.0.7)\n",
109"Collecting urllib3>=1.21.1\n",
110" Downloading urllib3-1.25.11-py2.py3-none-any.whl (127 kB)\n",
111"\u001b[K |████████████████████████████████| 127 kB 35.6 MB/s \n",
112"\u001b[?25hRequirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.7/dist-packages (from aiohttp->datasets) (21.4.0)\n",
113"Collecting multidict<7.0,>=4.5\n",
114" Downloading multidict-6.0.2-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (94 kB)\n",
115"\u001b[K |████████████████████████████████| 94 kB 3.2 MB/s \n",
116"\u001b[?25hCollecting yarl<2.0,>=1.0\n",
117" Downloading yarl-1.7.2-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl (271 kB)\n",
118"\u001b[K |████████████████████████████████| 271 kB 57.7 MB/s \n",
119"\u001b[?25hCollecting frozenlist>=1.1.1\n",
120" Downloading frozenlist-1.3.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_17_x86_64.manylinux2014_x86_64.whl (144 kB)\n",
121"\u001b[K |████████████████████████████████| 144 kB 74.6 MB/s \n",
122"\u001b[?25hCollecting aiosignal>=1.1.2\n",
123" Downloading aiosignal-1.2.0-py3-none-any.whl (8.2 kB)\n",
124"Collecting async-timeout<5.0,>=4.0.0a3\n",
125" Downloading async_timeout-4.0.2-py3-none-any.whl (5.8 kB)\n",
126"Requirement already satisfied: charset-normalizer<3.0,>=2.0 in /usr/local/lib/python3.7/dist-packages (from aiohttp->datasets) (2.0.12)\n",
127"Collecting asynctest==0.13.0\n",
128" Downloading asynctest-0.13.0-py3-none-any.whl (26 kB)\n",
129"Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->datasets) (3.7.0)\n",
130"Requirement already satisfied: pytz>=2017.3 in /usr/local/lib/python3.7/dist-packages (from pandas->datasets) (2018.9)\n",
131"Building wheels for collected packages: cohere\n",
132" Building wheel for cohere (setup.py) ... \u001b[?25l\u001b[?25hdone\n",
133" Created wheel for cohere: filename=cohere-1.3.2-py3-none-any.whl size=8756 sha256=7100be99e9471aad317473cdb89b5f8437262cd249338bf5fc3c6e2f7637207c\n",
134" Stored in directory: /root/.cache/pip/wheels/4b/65/39/565f9e56448829a12bcf708ead2e6b233ca2171cc113a59a7c\n",
135"Successfully built cohere\n",
136"Installing collected packages: multidict, frozenlist, yarl, urllib3, asynctest, async-timeout, aiosignal, pyyaml, fsspec, aiohttp, xxhash, responses, loguru, huggingface-hub, dnspython, pinecone-client, datasets, cohere\n",
137" Attempting uninstall: urllib3\n",
138" Found existing installation: urllib3 1.24.3\n",
139" Uninstalling urllib3-1.24.3:\n",
140" Successfully uninstalled urllib3-1.24.3\n",
141" Attempting uninstall: pyyaml\n",
142" Found existing installation: PyYAML 3.13\n",
143" Uninstalling PyYAML-3.13:\n",
144" Successfully uninstalled PyYAML-3.13\n",
145"\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n",
146"datascience 0.10.6 requires folium==0.2.1, but you have folium 0.8.3 which is incompatible.\u001b[0m\n",
147"Successfully installed aiohttp-3.8.1 aiosignal-1.2.0 async-timeout-4.0.2 asynctest-0.13.0 cohere-1.3.2 datasets-2.0.0 dnspython-2.2.1 frozenlist-1.3.0 fsspec-2022.2.0 huggingface-hub-0.4.0 loguru-0.6.0 multidict-6.0.2 pinecone-client-2.0.8 pyyaml-6.0 responses-0.18.0 urllib3-1.25.11 xxhash-3.0.0 yarl-1.7.2\n"
148]
149}
150],
151"source": [
152"!pip install cohere pinecone-client datasets"
153]
154},
155{
156"cell_type": "markdown",
157"metadata": {
158"id": "-aPF7S50NA1P"
159},
160"source": [
161"And sign up for an API key over at [Cohere](https://os.cohere.ai/) and [Pinecone](https://app.pinecone.io), you can enter the keys directly in the cell below."
162]
163},
164{
165"cell_type": "code",
166"execution_count": null,
167"metadata": {
168"id": "kYBByOHYNA1Q"
169},
170"outputs": [],
171"source": [
172"COHERE_KEY = '<<YOUR-KEY-HERE>>'\n",
173"PINECONE_KEY = '<<YOUR-KEY-HERE>>'"
174]
175},
176{
177"cell_type": "markdown",
178"metadata": {
179"id": "95KrmFq6NA1Q"
180},
181"source": [
182"## Create Embeddings"
183]
184},
185{
186"cell_type": "markdown",
187"metadata": {
188"id": "3yQxt8ukNA1R"
189},
190"source": [
191"We can create sentence embeddings easily using Cohere. First, we import the Cohere client and initialize our connection using the API key we retrieved earlier."
192]
193},
194{
195"cell_type": "code",
196"execution_count": null,
197"metadata": {
198"id": "BPllmdUWNA1R"
199},
200"outputs": [],
201"source": [
202"import cohere\n",
203"\n",
204"co = cohere.Client(COHERE_KEY)"
205]
206},
207{
208"cell_type": "markdown",
209"metadata": {
210"id": "UXRk3gMENA1R"
211},
212"source": [
213"We will load the **T**ext **RE**trieval **C**onference (TREC) question classification dataset which contains 5.5K labeled questions. We will take the first 1K samples for this demo, but this can be scaled to millions or even billions of samples."
214]
215},
216{
217"cell_type": "code",
218"execution_count": null,
219"metadata": {
220"colab": {
221"base_uri": "https://localhost:8080/",
222"height": 351,
223"referenced_widgets": [
224"b22413d2b4444523b39a3d52acaa8587",
225"1316d3504b9040c580aba3eb0a7d5e0e",
226"c821ecea97044016a2aaa9286a672a32",
227"029cb2f03eda4b13b800a1db177a3f60",
228"39b8642c3fbb4bc5b6f0be6202b62307",
229"4757ba7fe9b644faad80e3910978457d",
230"c7f29593c65d4df8b60e56374beebf02",
231"c91efde0092348888ebd08afb3dce9c3",
232"dbe7c2c262424b6ea04621a6b2e5281c",
233"2ac3ffc96b1348e5997cf6d3a78720cd",
234"12fa683ee05d42a3bea42cc6faa32708",
235"0d6d62d7332f42cd8e1ec57a663fea8a",
236"85fd7924781a40aab82e823e8e860909",
237"8fe2d4de448a4703b9ce6c52a5758eb4",
238"6aa7ba4ec9464067a5816869e62b67c2",
239"6de1c75f642d484b856b36ec49d68094",
240"a6cad1f5f5724757b62773d3e65f9cb3",
241"8975440fb35246f5b98085fc4f78f03f",
242"e4d008ff14684317a873c7607a1fec0a",
243"1c6cd51d6bed4f5fbb0993b957c22c84",
244"3415e17e7a524cdea62af74b89050530",
245"28a99f10e2eb42a5b8e86be49f7186dc",
246"bb0b95d8f00a4f1494df44bf0d2f7a26",
247"62b6a3221bb745eeb13403708f0487bf",
248"a27c9808eba4418aabf40a9e97356c4a",
249"ac6de5de78b24d45a17fdb5c800683b9",
250"bfebdd15ffff4e079e4939b9444ce467",
251"308d349564ec408b8e18f8486ed1de85",
252"4c3ed023db9b4cf1b6cbd6e763829232",
253"5c730e1a02104929bf61a227dfaea883",
254"f0dbac8f3ac4449db25e33fef3840665",
255"028f3dc9c8d148db99ec5df93b3c8773",
256"8a7c916f389541279a5791a4e4372cf5",
257"5b524ec90a634846a9de9b22171a036f",
258"be9f9665d9b4421387d466b36dc12e53",
259"e999e472f30646119a333aedcf1eb42c",
260"40b432b310e44844b8291471c1521f20",
261"b99172f8069349d0ab155f77b35ee6c6",
262"a57e5e6663944c478d949f2ec12ca51b",
263"0e9854e5ce154135b0aee7e5ea795857",
264"d4da7a1fc98f44d1bae157d3c9c07942",
265"c70aaeea77db4db6903c5873817300b0",
266"3e7b0222f5024a599892a17d78d13526",
267"7b19a341d5174298b74b822929fe3cb9",
268"3ecef000cd494737a7ba938505ac55de",
269"bd3edc0d7c174ec09a3b8a881a22cbaf",
270"a3b312ea3f3a4558a4e0db94b1561a64",
271"1ad595db69bd4f9caf30b46027bfde96",
272"8d9f9db2e8704eb18ca2a1920cb06ba2",
273"45dab7f0fc3b4e0e89e0a3193309b834",
274"d27c0648c05f40ef9e175d27135ec751",
275"ee7b63007cc3482aa89857c9bce99e22",
276"f51393b65613425e9f732b35f8e94e24",
277"2aa88dd5076049e1958bd8c7accbe5b0",
278"1788ec736d0343f3b9c1201c82362b7d",
279"060928a16cbd435b866624a6531b6013",
280"4239287597f344919e756848c5e1536a",
281"d959e84d5a9b40a998ce16bc4dc92fbd",
282"ac50c9ba79be4493a12213e01c140289",
283"41cf0a3151be4597bee89c3c6d3b9a27",
284"ea02fb4d9fbe451fabffa0001ff8a1a4",
285"c7e26ccc3f574a39b80553fdb9c3b70e",
286"adcaf0a904a54127b212ca3751f9f95b",
287"261c85699616441783c044118d9e165f",
288"0a93329af24246f4b77acf0e5574b335",
289"1f6e8e2374bf4b3d99727eb9ed157cb6",
290"ea5454b5334b40bea6d179a675d6578e",
291"f6b507d2bdb74ebaadbbd8a9b51290d6",
292"f9e79559b8e34d55b669413423c3c2ae",
293"13d8f25b01ad4f9fb12203ee85f26d50",
294"2e7a3ef5cfa34370bffe45fddb73ae8d",
295"32b6df9a4f2b42739355c896cea4ca4f",
296"e532f8f1a6cf433fa22410a6876f4707",
297"f5632381613a400f88d61d520f7f27e7",
298"6b9659cabd534ecb983fc84da293cfdc",
299"f8cbbb20284f43ea9efcaefc8a790f9a",
300"f2d8146acd2a43298266cf4649ef8a68",
301"d87354f9ad594885903943925d186ae0",
302"21e580ef7de94f068a12c9bd4cce14d6",
303"2dfea83397ce4cb78dc4b14d7d19f00f",
304"dc8bbd3f9ae44ccbb53d5ccf33254f14",
305"c2b5da2a0bf64686b1e9c3784bc7ba48",
306"bf415d781ee84e2fb940b64d2d0490b0",
307"62f17160914242c0a2ce825816be36c8",
308"82536a1182ca49cb8fc103045706642f",
309"c2c9811d883a4a619e900fd8e66bde8b",
310"7ebfff0fd00f43a4a3c68b6955894e24",
311"114dcbb9bdf3404db0518ee34e9ab7fc"
312]
313},
314"id": "U7qOsNCLNA1S",
315"outputId": "31c12f61-f3d7-44a7-8524-eb4c9c203ada"
316},
317"outputs": [
318{
319"data": {
320"application/vnd.jupyter.widget-view+json": {
321"model_id": "b22413d2b4444523b39a3d52acaa8587",
322"version_major": 2,
323"version_minor": 0
324},
325"text/plain": [
326"Downloading builder script: 0%| | 0.00/2.22k [00:00<?, ?B/s]"
327]
328},
329"metadata": {},
330"output_type": "display_data"
331},
332{
333"data": {
334"application/vnd.jupyter.widget-view+json": {
335"model_id": "0d6d62d7332f42cd8e1ec57a663fea8a",
336"version_major": 2,
337"version_minor": 0
338},
339"text/plain": [
340"Downloading metadata: 0%| | 0.00/1.42k [00:00<?, ?B/s]"
341]
342},
343"metadata": {},
344"output_type": "display_data"
345},
346{
347"name": "stderr",
348"output_type": "stream",
349"text": [
350"Using custom data configuration default\n"
351]
352},
353{
354"name": "stdout",
355"output_type": "stream",
356"text": [
357"Downloading and preparing dataset trec/default (download: 350.79 KiB, generated: 403.39 KiB, post-processed: Unknown size, total: 754.18 KiB) to /root/.cache/huggingface/datasets/trec/default/1.1.0/751da1ab101b8d297a3d6e9c79ee9b0173ff94c4497b75677b59b61d5467a9b9...\n"
358]
359},
360{
361"data": {
362"application/vnd.jupyter.widget-view+json": {
363"model_id": "bb0b95d8f00a4f1494df44bf0d2f7a26",
364"version_major": 2,
365"version_minor": 0
366},
367"text/plain": [
368"Downloading data files: 0%| | 0/2 [00:00<?, ?it/s]"
369]
370},
371"metadata": {},
372"output_type": "display_data"
373},
374{
375"data": {
376"application/vnd.jupyter.widget-view+json": {
377"model_id": "5b524ec90a634846a9de9b22171a036f",
378"version_major": 2,
379"version_minor": 0
380},
381"text/plain": [
382"Downloading data: 0%| | 0.00/336k [00:00<?, ?B/s]"
383]
384},
385"metadata": {},
386"output_type": "display_data"
387},
388{
389"data": {
390"application/vnd.jupyter.widget-view+json": {
391"model_id": "3ecef000cd494737a7ba938505ac55de",
392"version_major": 2,
393"version_minor": 0
394},
395"text/plain": [
396"Downloading data: 0%| | 0.00/23.4k [00:00<?, ?B/s]"
397]
398},
399"metadata": {},
400"output_type": "display_data"
401},
402{
403"data": {
404"application/vnd.jupyter.widget-view+json": {
405"model_id": "060928a16cbd435b866624a6531b6013",
406"version_major": 2,
407"version_minor": 0
408},
409"text/plain": [
410"Extracting data files: 0%| | 0/2 [00:00<?, ?it/s]"
411]
412},
413"metadata": {},
414"output_type": "display_data"
415},
416{
417"data": {
418"application/vnd.jupyter.widget-view+json": {
419"model_id": "ea5454b5334b40bea6d179a675d6578e",
420"version_major": 2,
421"version_minor": 0
422},
423"text/plain": [
424"Generating train split: 0%| | 0/5452 [00:00<?, ? examples/s]"
425]
426},
427"metadata": {},
428"output_type": "display_data"
429},
430{
431"data": {
432"application/vnd.jupyter.widget-view+json": {
433"model_id": "d87354f9ad594885903943925d186ae0",
434"version_major": 2,
435"version_minor": 0
436},
437"text/plain": [
438"Generating test split: 0%| | 0/500 [00:00<?, ? examples/s]"
439]
440},
441"metadata": {},
442"output_type": "display_data"
443},
444{
445"name": "stdout",
446"output_type": "stream",
447"text": [
448"Dataset trec downloaded and prepared to /root/.cache/huggingface/datasets/trec/default/1.1.0/751da1ab101b8d297a3d6e9c79ee9b0173ff94c4497b75677b59b61d5467a9b9. Subsequent calls will reuse this data.\n"
449]
450},
451{
452"data": {
453"text/plain": [
454"Dataset({\n",
455" features: ['label-coarse', 'label-fine', 'text'],\n",
456" num_rows: 1000\n",
457"})"
458]
459},
460"execution_count": 4,
461"metadata": {},
462"output_type": "execute_result"
463}
464],
465"source": [
466"from datasets import load_dataset\n",
467"\n",
468"# load the first 1K rows of the TREC dataset\n",
469"trec = load_dataset('trec', split='train[:1000]')\n",
470"trec"
471]
472},
473{
474"cell_type": "code",
475"execution_count": null,
476"metadata": {
477"colab": {
478"base_uri": "https://localhost:8080/"
479},
480"id": "lKGiLkWpNA1S",
481"outputId": "0e09eaf7-10ff-4081-fa0d-41bdd46f4b87"
482},
483"outputs": [
484{
485"data": {
486"text/plain": [
487"{'label-coarse': 0,\n",
488" 'label-fine': 0,\n",
489" 'text': 'How did serfdom develop in and then leave Russia ?'}"
490]
491},
492"execution_count": 5,
493"metadata": {},
494"output_type": "execute_result"
495}
496],
497"source": [
498"trec[0]"
499]
500},
501{
502"cell_type": "markdown",
503"metadata": {
504"id": "kQc5RPosNA1T"
505},
506"source": [
507"We can then pass these questions to Cohere to create embeddings."
508]
509},
510{
511"cell_type": "code",
512"execution_count": null,
513"metadata": {
514"id": "xfAT8ojCNA1T"
515},
516"outputs": [],
517"source": [
518"embeds = co.embed(\n",
519" texts=trec['text'],\n",
520" model='small',\n",
521" truncate='LEFT'\n",
522").embeddings"
523]
524},
525{
526"cell_type": "markdown",
527"metadata": {
528"id": "ojqPN8ZjNA1T"
529},
530"source": [
531"We can check the dimensionality of the returned vectors, for this we will convert it from a list of lists to a Numpy array. We will need to save the embedding dimensionality from this to be used when initializing our Pinecone index later."
532]
533},
534{
535"cell_type": "code",
536"execution_count": null,
537"metadata": {
538"colab": {
539"base_uri": "https://localhost:8080/"
540},
541"id": "5_7NKhUeNA1T",
542"outputId": "38172d37-3774-46b6-962f-6672bbab1ab1"
543},
544"outputs": [
545{
546"data": {
547"text/plain": [
548"(1000, 1024)"
549]
550},
551"execution_count": 7,
552"metadata": {},
553"output_type": "execute_result"
554}
555],
556"source": [
557"import numpy as np\n",
558"\n",
559"shape = np.array(embeds).shape\n",
560"shape"
561]
562},
563{
564"cell_type": "markdown",
565"metadata": {
566"id": "tJoX68NENA1T"
567},
568"source": [
569"Here we can see the `1024` embedding dimensionality produced by Cohere's small model, and the `1000` samples we built embeddings for."
570]
571},
572{
573"cell_type": "markdown",
574"metadata": {
575"id": "enzeBYDONA1U"
576},
577"source": [
578"## Storing the Embeddings"
579]
580},
581{
582"cell_type": "markdown",
583"metadata": {
584"id": "E5I5jsQwNA1U"
585},
586"source": [
587"Now that we have our embeddings we can move on to indexing them in the Pinecone vector database. Again, this is very simple, we just initialize our connection to Pinecone and then create a new index for storing the embeddings, making sure to specify that we would like to use the cosine similarity metric to align with Cohere's embeddings."
588]
589},
590{
591"cell_type": "code",
592"execution_count": null,
593"metadata": {
594"id": "iwNgMCVgNA1U"
595},
596"outputs": [],
597"source": [
598"from pinecone import Pinecone\n",
599"\n",
600"pinecone.init(\n",
601" PINECONE_KEY,\n",
602" environment=\"YOUR_ENV\" # find next to API key in console\n",
603")\n",
604"\n",
605"index_name = 'cohere-pinecone-trec'\n",
606"\n",
607"# if the index does not exist, we create it\n",
608"if index_name not in pinecone.list_indexes().names():\n",
609" pinecone.create_index(\n",
610" index_name,\n",
611" dimension=shape[1],\n",
612" metric='cosine'\n",
613" )\n",
614"\n",
615"# connect to index\n",
616"index = pinecone.Index(index_name)"
617]
618},
619{
620"cell_type": "markdown",
621"metadata": {
622"id": "f20oKqIMNA1U"
623},
624"source": [
625"Now we can begin populating the index with our embeddings. Pinecone expects us to provide a list of tuples in the format *(id, vector, metadata)*, where the *metadata* field is an optional extra field where we can store anything we want in a dictionary format. For this example, we will store the original text of the embeddings.\n",
626"\n",
627"While uploading our data, we will batch everything to avoid pushing too much data in one go."
628]
629},
630{
631"cell_type": "code",
632"execution_count": null,
633"metadata": {
634"colab": {
635"base_uri": "https://localhost:8080/"
636},
637"id": "QO1U0uvZNA1U",
638"outputId": "eecb00bb-dbfc-4331-ef90-8a53eb7e303d"
639},
640"outputs": [
641{
642"data": {
643"text/plain": [
644"{'dimension': 1024,\n",
645" 'index_fullness': 0.0,\n",
646" 'namespaces': {'': {'vector_count': 1000}}}"
647]
648},
649"execution_count": 9,
650"metadata": {},
651"output_type": "execute_result"
652}
653],
654"source": [
655"batch_size = 128\n",
656"\n",
657"ids = [str(i) for i in range(shape[0])]\n",
658"# create list of metadata dictionaries\n",
659"meta = [{'text': text} for text in trec['text']]\n",
660"\n",
661"# create list of (id, vector, metadata) tuples to be upserted\n",
662"to_upsert = list(zip(ids, embeds, meta))\n",
663"\n",
664"for i in range(0, shape[0], batch_size):\n",
665" i_end = min(i+batch_size, shape[0])\n",
666" index.upsert(vectors=to_upsert[i:i_end])\n",
667"\n",
668"# let's view the index statistics\n",
669"index.describe_index_stats()"
670]
671},
672{
673"cell_type": "markdown",
674"metadata": {
675"id": "49rMcQEeNA1U"
676},
677"source": [
678"Perfect, we can see from `index.describe_index_stats` that we have a *1024-dimensionality* index populated with *1000* embeddings. The `indexFullness` metric tells us how full our index is, at the moment it is empty. Using the default value of one *p1* pod we can fit ~750K embeddings before the `indexFullness` reaches capacity. The [Usage Estimator](www.pinecone.io/pricing) can be used to identify the number of pods required for a given number of *n*-dimensional embeddings."
679]
680},
681{
682"cell_type": "markdown",
683"metadata": {
684"id": "4r3G0L9MNA1V"
685},
686"source": [
687"## Semantic Search"
688]
689},
690{
691"cell_type": "markdown",
692"metadata": {
693"id": "wqQ6-UVHNA1V"
694},
695"source": [
696"Now that we have our indexed vectors we can perform a few search queries. When searching we will first embed our query using Cohere, and then search using the returned vector in Pinecone."
697]
698},
699{
700"cell_type": "code",
701"execution_count": null,
702"metadata": {
703"colab": {
704"base_uri": "https://localhost:8080/"
705},
706"id": "uEq-2XDMNA1V",
707"outputId": "06099cf6-2527-43fe-9514-d6a7657a6c5f"
708},
709"outputs": [
710{
711"name": "stdout",
712"output_type": "stream",
713"text": [
714"(1, 1024)\n"
715]
716},
717{
718"data": {
719"text/plain": [
720"{'results': [{'matches': [{'id': '932',\n",
721" 'metadata': {'text': 'Why did the world enter a '\n",
722" 'global depression in 1929 ?'},\n",
723" 'score': 0.832818151,\n",
724" 'values': []},\n",
725" {'id': '787',\n",
726" 'metadata': {'text': 'When was `` the Great '\n",
727" \"Depression '' ?\"},\n",
728" 'score': 0.752612948,\n",
729" 'values': []},\n",
730" {'id': '400',\n",
731" 'metadata': {'text': 'What crop failure caused the '\n",
732" 'Irish Famine ?'},\n",
733" 'score': 0.499015927,\n",
734" 'values': []},\n",
735" {'id': '160',\n",
736" 'metadata': {'text': 'What war did the '\n",
737" 'Wanna-Go-Home Riots occur '\n",
738" 'after ?'},\n",
739" 'score': 0.339323878,\n",
740" 'values': []},\n",
741" {'id': '835',\n",
742" 'metadata': {'text': 'What were popular songs and '\n",
743" 'types of songs in the 1920s '\n",
744" '?'},\n",
745" 'score': 0.337981969,\n",
746" 'values': []},\n",
747" {'id': '481',\n",
748" 'metadata': {'text': 'What caused the Lynmouth '\n",
749" 'floods ?'},\n",
750" 'score': 0.337701261,\n",
751" 'values': []},\n",
752" {'id': '864',\n",
753" 'metadata': {'text': 'When did the Dow first reach '\n",
754" '?'},\n",
755" 'score': 0.326343566,\n",
756" 'values': []},\n",
757" {'id': '14',\n",
758" 'metadata': {'text': 'What is considered the '\n",
759" 'costliest disaster the '\n",
760" 'insurance industry has ever '\n",
761" 'faced ?'},\n",
762" 'score': 0.317433923,\n",
763" 'values': []},\n",
764" {'id': '262',\n",
765" 'metadata': {'text': 'When did World War I start ?'},\n",
766" 'score': 0.317194521,\n",
767" 'values': []},\n",
768" {'id': '233',\n",
769" 'metadata': {'text': \"What caused Harry Houdini 's \"\n",
770" 'death ?'},\n",
771" 'score': 0.308083236,\n",
772" 'values': []}],\n",
773" 'namespace': ''}]}"
774]
775},
776"execution_count": 10,
777"metadata": {},
778"output_type": "execute_result"
779}
780],
781"source": [
782"query = \"What caused the 1929 Great Depression?\"\n",
783"\n",
784"# create the query embedding\n",
785"xq = co.embed(\n",
786" texts=[query],\n",
787" model='small',\n",
788" truncate='LEFT'\n",
789").embeddings\n",
790"\n",
791"print(np.array(xq).shape)\n",
792"\n",
793"# query, returning the top 10 most similar results\n",
794"res = index.query(vector=xq, top_k=10, include_metadata=True)\n",
795"res"
796]
797},
798{
799"cell_type": "markdown",
800"metadata": {
801"id": "ve68DEOUNA1V"
802},
803"source": [
804"The response from Pinecone includes our original text in the `metadata` field, let's print out the `top_k` most similar questions and their respective similarity scores."
805]
806},
807{
808"cell_type": "code",
809"execution_count": null,
810"metadata": {
811"colab": {
812"base_uri": "https://localhost:8080/"
813},
814"id": "-X3jpU3TNA1V",
815"outputId": "fb27390c-3bed-4f26-8421-f23812a82df7"
816},
817"outputs": [
818{
819"name": "stdout",
820"output_type": "stream",
821"text": [
822"0.83: Why did the world enter a global depression in 1929 ?\n",
823"0.75: When was `` the Great Depression '' ?\n",
824"0.50: What crop failure caused the Irish Famine ?\n",
825"0.34: What war did the Wanna-Go-Home Riots occur after ?\n",
826"0.34: What were popular songs and types of songs in the 1920s ?\n",
827"0.34: What caused the Lynmouth floods ?\n",
828"0.33: When did the Dow first reach ?\n",
829"0.32: What is considered the costliest disaster the insurance industry has ever faced ?\n",
830"0.32: When did World War I start ?\n",
831"0.31: What caused Harry Houdini 's death ?\n"
832]
833}
834],
835"source": [
836"for match in res['matches']:\n",
837" print(f\"{match['score']:.2f}: {match['metadata']['text']}\")"
838]
839},
840{
841"cell_type": "markdown",
842"metadata": {
843"id": "0yKEvtGWNA1V"
844},
845"source": [
846"Looks good, let's make it harder and replace *\"depression\"* with the incorrect term *\"recession\"*."
847]
848},
849{
850"cell_type": "code",
851"execution_count": null,
852"metadata": {
853"colab": {
854"base_uri": "https://localhost:8080/"
855},
856"id": "HY5NSwiGNA1V",
857"outputId": "3aa10ef1-beff-47a8-f468-b66a8a5b1b5a"
858},
859"outputs": [
860{
861"name": "stdout",
862"output_type": "stream",
863"text": [
864"0.66: Why did the world enter a global depression in 1929 ?\n",
865"0.61: When was `` the Great Depression '' ?\n",
866"0.43: What are some of the significant historical events of the 1990s ?\n",
867"0.43: What crop failure caused the Irish Famine ?\n",
868"0.37: What were popular songs and types of songs in the 1920s ?\n",
869"0.36: When did the Dow first reach ?\n",
870"0.35: What war did the Wanna-Go-Home Riots occur after ?\n",
871"0.34: What historical event happened in Dogtown in 1899 ?\n",
872"0.33: What is considered the costliest disaster the insurance industry has ever faced ?\n",
873"0.31: What was the education system in the 1960 's ?\n"
874]
875}
876],
877"source": [
878"query = \"What was the cause of the major recession in the early 20th century?\"\n",
879"\n",
880"# create the query embedding\n",
881"xq = co.embed(\n",
882" texts=[query],\n",
883" model='small',\n",
884" truncate='LEFT'\n",
885").embeddings\n",
886"\n",
887"# query, returning the top 10 most similar results\n",
888"res = index.query(vector=xq, top_k=10, include_metadata=True)\n",
889"\n",
890"for match in res['matches']:\n",
891" print(f\"{match['score']:.2f}: {match['metadata']['text']}\")"
892]
893},
894{
895"cell_type": "markdown",
896"metadata": {
897"id": "y6aPDXFTNA1V"
898},
899"source": [
900"And again."
901]
902},
903{
904"cell_type": "code",
905"execution_count": null,
906"metadata": {
907"colab": {
908"base_uri": "https://localhost:8080/"
909},
910"id": "EQYU6_bkNA1V",
911"outputId": "46e5a4e9-e136-436a-9136-3e74075e98b1"
912},
913"outputs": [
914{
915"name": "stdout",
916"output_type": "stream",
917"text": [
918"0.71: Why did the world enter a global depression in 1929 ?\n",
919"0.62: When was `` the Great Depression '' ?\n",
920"0.40: What crop failure caused the Irish Famine ?\n",
921"0.38: What are some of the significant historical events of the 1990s ?\n",
922"0.38: When did the Dow first reach ?\n",
923"0.35: What were popular songs and types of songs in the 1920s ?\n",
924"0.33: What was the education system in the 1960 's ?\n",
925"0.32: Give a reason for American Indians oftentimes dropping out of school .\n",
926"0.31: What war did the Wanna-Go-Home Riots occur after ?\n",
927"0.30: What historical event happened in Dogtown in 1899 ?\n"
928]
929}
930],
931"source": [
932"query = \"Why was there a long-term economic downturn in the early 20th century?\"\n",
933"\n",
934"# create the query embedding\n",
935"xq = co.embed(\n",
936" texts=[query],\n",
937" model='small',\n",
938" truncate='LEFT'\n",
939").embeddings\n",
940"\n",
941"# query, returning the top 10 most similar results\n",
942"res = index.query(vector=xq, top_k=10, include_metadata=True)\n",
943"\n",
944"for match in res['matches']:\n",
945" print(f\"{match['score']:.2f}: {match['metadata']['text']}\")"
946]
947},
948{
949"attachments": {},
950"cell_type": "markdown",
951"metadata": {},
952"source": [
953"Looks great, our semantic search pipeline is clearly able to identify the meaning between each of our queries and return the most semantically similar questions from the already indexed questions.\n",
954"\n",
955"Once we're done with the index we delete it to save resources:"
956]
957},
958{
959"cell_type": "code",
960"execution_count": null,
961"metadata": {},
962"outputs": [],
963"source": [
964"piencone.delete_index(index_name)"
965]
966},
967{
968"attachments": {},
969"cell_type": "markdown",
970"metadata": {
971"id": "gR-oj6C9NA1W"
972},
973"source": [
974"---"
975]
976}
977],
978"metadata": {
979"colab": {
980"collapsed_sections": [],
981"include_colab_link": true,
982"name": "semantic-search-trec.ipynb",
983"provenance": []
984},
985"kernelspec": {
986"display_name": "Python 3",
987"language": "python",
988"name": "python3"
989},
990"language_info": {
991"codemirror_mode": {
992"name": "ipython",
993"version": 3
994},
995"file_extension": ".py",
996"mimetype": "text/x-python",
997"name": "python",
998"nbconvert_exporter": "python",
999"pygments_lexer": "ipython3",
1000"version": "3.10.7 (main, Sep 14 2022, 22:38:23) [Clang 14.0.0 (clang-1400.0.29.102)]"
1001},
1002"orig_nbformat": 4,
1003"vscode": {
1004"interpreter": {
1005"hash": "b0fa6594d8f4cbf19f97940f81e996739fb7646882a419484c72d19e05852a7e"
1006}
1007},
1008"widgets": {
1009"application/vnd.jupyter.widget-state+json": {
1010"028f3dc9c8d148db99ec5df93b3c8773": {
1011"model_module": "@jupyter-widgets/base",
1012"model_module_version": "1.2.0",
1013"model_name": "LayoutModel",
1014"state": {
1015"_model_module": "@jupyter-widgets/base",
1016"_model_module_version": "1.2.0",
1017"_model_name": "LayoutModel",
1018"_view_count": null,
1019"_view_module": "@jupyter-widgets/base",
1020"_view_module_version": "1.2.0",
1021"_view_name": "LayoutView",
1022"align_content": null,
1023"align_items": null,
1024"align_self": null,
1025"border": null,
1026"bottom": null,
1027"display": null,
1028"flex": null,
1029"flex_flow": null,
1030"grid_area": null,
1031"grid_auto_columns": null,
1032"grid_auto_flow": null,
1033"grid_auto_rows": null,
1034"grid_column": null,
1035"grid_gap": null,
1036"grid_row": null,
1037"grid_template_areas": null,
1038"grid_template_columns": null,
1039"grid_template_rows": null,
1040"height": null,
1041"justify_content": null,
1042"justify_items": null,
1043"left": null,
1044"margin": null,
1045"max_height": null,
1046"max_width": null,
1047"min_height": null,
1048"min_width": null,
1049"object_fit": null,
1050"object_position": null,
1051"order": null,
1052"overflow": null,
1053"overflow_x": null,
1054"overflow_y": null,
1055"padding": null,
1056"right": null,
1057"top": null,
1058"visibility": null,
1059"width": null
1060}
1061},
1062"029cb2f03eda4b13b800a1db177a3f60": {
1063"model_module": "@jupyter-widgets/controls",
1064"model_module_version": "1.5.0",
1065"model_name": "HTMLModel",
1066"state": {
1067"_dom_classes": [],
1068"_model_module": "@jupyter-widgets/controls",
1069"_model_module_version": "1.5.0",
1070"_model_name": "HTMLModel",
1071"_view_count": null,
1072"_view_module": "@jupyter-widgets/controls",
1073"_view_module_version": "1.5.0",
1074"_view_name": "HTMLView",
1075"description": "",
1076"description_tooltip": null,
1077"layout": "IPY_MODEL_2ac3ffc96b1348e5997cf6d3a78720cd",
1078"placeholder": "",
1079"style": "IPY_MODEL_12fa683ee05d42a3bea42cc6faa32708",
1080"value": " 5.50k/? [00:00<00:00, 113kB/s]"
1081}
1082},
1083"060928a16cbd435b866624a6531b6013": {
1084"model_module": "@jupyter-widgets/controls",
1085"model_module_version": "1.5.0",
1086"model_name": "HBoxModel",
1087"state": {
1088"_dom_classes": [],
1089"_model_module": "@jupyter-widgets/controls",
1090"_model_module_version": "1.5.0",
1091"_model_name": "HBoxModel",
1092"_view_count": null,
1093"_view_module": "@jupyter-widgets/controls",
1094"_view_module_version": "1.5.0",
1095"_view_name": "HBoxView",
1096"box_style": "",
1097"children": [
1098"IPY_MODEL_4239287597f344919e756848c5e1536a",
1099"IPY_MODEL_d959e84d5a9b40a998ce16bc4dc92fbd",
1100"IPY_MODEL_ac50c9ba79be4493a12213e01c140289"
1101],
1102"layout": "IPY_MODEL_41cf0a3151be4597bee89c3c6d3b9a27"
1103}
1104},
1105"0a93329af24246f4b77acf0e5574b335": {
1106"model_module": "@jupyter-widgets/base",
1107"model_module_version": "1.2.0",
1108"model_name": "LayoutModel",
1109"state": {
1110"_model_module": "@jupyter-widgets/base",
1111"_model_module_version": "1.2.0",
1112"_model_name": "LayoutModel",
1113"_view_count": null,
1114"_view_module": "@jupyter-widgets/base",
1115"_view_module_version": "1.2.0",
1116"_view_name": "LayoutView",
1117"align_content": null,
1118"align_items": null,
1119"align_self": null,
1120"border": null,
1121"bottom": null,
1122"display": null,
1123"flex": null,
1124"flex_flow": null,
1125"grid_area": null,
1126"grid_auto_columns": null,
1127"grid_auto_flow": null,
1128"grid_auto_rows": null,
1129"grid_column": null,
1130"grid_gap": null,
1131"grid_row": null,
1132"grid_template_areas": null,
1133"grid_template_columns": null,
1134"grid_template_rows": null,
1135"height": null,
1136"justify_content": null,
1137"justify_items": null,
1138"left": null,
1139"margin": null,
1140"max_height": null,
1141"max_width": null,
1142"min_height": null,
1143"min_width": null,
1144"object_fit": null,
1145"object_position": null,
1146"order": null,
1147"overflow": null,
1148"overflow_x": null,
1149"overflow_y": null,
1150"padding": null,
1151"right": null,
1152"top": null,
1153"visibility": null,
1154"width": null
1155}
1156},
1157"0d6d62d7332f42cd8e1ec57a663fea8a": {
1158"model_module": "@jupyter-widgets/controls",
1159"model_module_version": "1.5.0",
1160"model_name": "HBoxModel",
1161"state": {
1162"_dom_classes": [],
1163"_model_module": "@jupyter-widgets/controls",
1164"_model_module_version": "1.5.0",
1165"_model_name": "HBoxModel",
1166"_view_count": null,
1167"_view_module": "@jupyter-widgets/controls",
1168"_view_module_version": "1.5.0",
1169"_view_name": "HBoxView",
1170"box_style": "",
1171"children": [
1172"IPY_MODEL_85fd7924781a40aab82e823e8e860909",
1173"IPY_MODEL_8fe2d4de448a4703b9ce6c52a5758eb4",
1174"IPY_MODEL_6aa7ba4ec9464067a5816869e62b67c2"
1175],
1176"layout": "IPY_MODEL_6de1c75f642d484b856b36ec49d68094"
1177}
1178},
1179"0e9854e5ce154135b0aee7e5ea795857": {
1180"model_module": "@jupyter-widgets/controls",
1181"model_module_version": "1.5.0",
1182"model_name": "DescriptionStyleModel",
1183"state": {
1184"_model_module": "@jupyter-widgets/controls",
1185"_model_module_version": "1.5.0",
1186"_model_name": "DescriptionStyleModel",
1187"_view_count": null,
1188"_view_module": "@jupyter-widgets/base",
1189"_view_module_version": "1.2.0",
1190"_view_name": "StyleView",
1191"description_width": ""
1192}
1193},
1194"114dcbb9bdf3404db0518ee34e9ab7fc": {
1195"model_module": "@jupyter-widgets/controls",
1196"model_module_version": "1.5.0",
1197"model_name": "DescriptionStyleModel",
1198"state": {
1199"_model_module": "@jupyter-widgets/controls",
1200"_model_module_version": "1.5.0",
1201"_model_name": "DescriptionStyleModel",
1202"_view_count": null,
1203"_view_module": "@jupyter-widgets/base",
1204"_view_module_version": "1.2.0",
1205"_view_name": "StyleView",
1206"description_width": ""
1207}
1208},
1209"12fa683ee05d42a3bea42cc6faa32708": {
1210"model_module": "@jupyter-widgets/controls",
1211"model_module_version": "1.5.0",
1212"model_name": "DescriptionStyleModel",
1213"state": {
1214"_model_module": "@jupyter-widgets/controls",
1215"_model_module_version": "1.5.0",
1216"_model_name": "DescriptionStyleModel",
1217"_view_count": null,
1218"_view_module": "@jupyter-widgets/base",
1219"_view_module_version": "1.2.0",
1220"_view_name": "StyleView",
1221"description_width": ""
1222}
1223},
1224"1316d3504b9040c580aba3eb0a7d5e0e": {
1225"model_module": "@jupyter-widgets/controls",
1226"model_module_version": "1.5.0",
1227"model_name": "HTMLModel",
1228"state": {
1229"_dom_classes": [],
1230"_model_module": "@jupyter-widgets/controls",
1231"_model_module_version": "1.5.0",
1232"_model_name": "HTMLModel",
1233"_view_count": null,
1234"_view_module": "@jupyter-widgets/controls",
1235"_view_module_version": "1.5.0",
1236"_view_name": "HTMLView",
1237"description": "",
1238"description_tooltip": null,
1239"layout": "IPY_MODEL_4757ba7fe9b644faad80e3910978457d",
1240"placeholder": "",
1241"style": "IPY_MODEL_c7f29593c65d4df8b60e56374beebf02",
1242"value": "Downloading builder script: "
1243}
1244},
1245"13d8f25b01ad4f9fb12203ee85f26d50": {
1246"model_module": "@jupyter-widgets/controls",
1247"model_module_version": "1.5.0",
1248"model_name": "HTMLModel",
1249"state": {
1250"_dom_classes": [],
1251"_model_module": "@jupyter-widgets/controls",
1252"_model_module_version": "1.5.0",
1253"_model_name": "HTMLModel",
1254"_view_count": null,
1255"_view_module": "@jupyter-widgets/controls",
1256"_view_module_version": "1.5.0",
1257"_view_name": "HTMLView",
1258"description": "",
1259"description_tooltip": null,
1260"layout": "IPY_MODEL_f8cbbb20284f43ea9efcaefc8a790f9a",
1261"placeholder": "",
1262"style": "IPY_MODEL_f2d8146acd2a43298266cf4649ef8a68",
1263"value": " 5291/5452 [00:00<00:00, 19978.94 examples/s]"
1264}
1265},
1266"1788ec736d0343f3b9c1201c82362b7d": {
1267"model_module": "@jupyter-widgets/controls",
1268"model_module_version": "1.5.0",
1269"model_name": "DescriptionStyleModel",
1270"state": {
1271"_model_module": "@jupyter-widgets/controls",
1272"_model_module_version": "1.5.0",
1273"_model_name": "DescriptionStyleModel",
1274"_view_count": null,
1275"_view_module": "@jupyter-widgets/base",
1276"_view_module_version": "1.2.0",
1277"_view_name": "StyleView",
1278"description_width": ""
1279}
1280},
1281"1ad595db69bd4f9caf30b46027bfde96": {
1282"model_module": "@jupyter-widgets/controls",
1283"model_module_version": "1.5.0",
1284"model_name": "HTMLModel",
1285"state": {
1286"_dom_classes": [],
1287"_model_module": "@jupyter-widgets/controls",
1288"_model_module_version": "1.5.0",
1289"_model_name": "HTMLModel",
1290"_view_count": null,
1291"_view_module": "@jupyter-widgets/controls",
1292"_view_module_version": "1.5.0",
1293"_view_name": "HTMLView",
1294"description": "",
1295"description_tooltip": null,
1296"layout": "IPY_MODEL_2aa88dd5076049e1958bd8c7accbe5b0",
1297"placeholder": "",
1298"style": "IPY_MODEL_1788ec736d0343f3b9c1201c82362b7d",
1299"value": " 23.4k/23.4k [00:00<00:00, 339kB/s]"
1300}
1301},
1302"1c6cd51d6bed4f5fbb0993b957c22c84": {
1303"model_module": "@jupyter-widgets/controls",
1304"model_module_version": "1.5.0",
1305"model_name": "ProgressStyleModel",
1306"state": {
1307"_model_module": "@jupyter-widgets/controls",
1308"_model_module_version": "1.5.0",
1309"_model_name": "ProgressStyleModel",
1310"_view_count": null,
1311"_view_module": "@jupyter-widgets/base",
1312"_view_module_version": "1.2.0",
1313"_view_name": "StyleView",
1314"bar_color": null,
1315"description_width": ""
1316}
1317},
1318"1f6e8e2374bf4b3d99727eb9ed157cb6": {
1319"model_module": "@jupyter-widgets/controls",
1320"model_module_version": "1.5.0",
1321"model_name": "DescriptionStyleModel",
1322"state": {
1323"_model_module": "@jupyter-widgets/controls",
1324"_model_module_version": "1.5.0",
1325"_model_name": "DescriptionStyleModel",
1326"_view_count": null,
1327"_view_module": "@jupyter-widgets/base",
1328"_view_module_version": "1.2.0",
1329"_view_name": "StyleView",
1330"description_width": ""
1331}
1332},
1333"21e580ef7de94f068a12c9bd4cce14d6": {
1334"model_module": "@jupyter-widgets/controls",
1335"model_module_version": "1.5.0",
1336"model_name": "HTMLModel",
1337"state": {
1338"_dom_classes": [],
1339"_model_module": "@jupyter-widgets/controls",
1340"_model_module_version": "1.5.0",
1341"_model_name": "HTMLModel",
1342"_view_count": null,
1343"_view_module": "@jupyter-widgets/controls",
1344"_view_module_version": "1.5.0",
1345"_view_name": "HTMLView",
1346"description": "",
1347"description_tooltip": null,
1348"layout": "IPY_MODEL_bf415d781ee84e2fb940b64d2d0490b0",
1349"placeholder": "",
1350"style": "IPY_MODEL_62f17160914242c0a2ce825816be36c8",
1351"value": "Generating test split: 0%"
1352}
1353},
1354"261c85699616441783c044118d9e165f": {
1355"model_module": "@jupyter-widgets/controls",
1356"model_module_version": "1.5.0",
1357"model_name": "ProgressStyleModel",
1358"state": {
1359"_model_module": "@jupyter-widgets/controls",
1360"_model_module_version": "1.5.0",
1361"_model_name": "ProgressStyleModel",
1362"_view_count": null,
1363"_view_module": "@jupyter-widgets/base",
1364"_view_module_version": "1.2.0",
1365"_view_name": "StyleView",
1366"bar_color": null,
1367"description_width": ""
1368}
1369},
1370"28a99f10e2eb42a5b8e86be49f7186dc": {
1371"model_module": "@jupyter-widgets/controls",
1372"model_module_version": "1.5.0",
1373"model_name": "DescriptionStyleModel",
1374"state": {
1375"_model_module": "@jupyter-widgets/controls",
1376"_model_module_version": "1.5.0",
1377"_model_name": "DescriptionStyleModel",
1378"_view_count": null,
1379"_view_module": "@jupyter-widgets/base",
1380"_view_module_version": "1.2.0",
1381"_view_name": "StyleView",
1382"description_width": ""
1383}
1384},
1385"2aa88dd5076049e1958bd8c7accbe5b0": {
1386"model_module": "@jupyter-widgets/base",
1387"model_module_version": "1.2.0",
1388"model_name": "LayoutModel",
1389"state": {
1390"_model_module": "@jupyter-widgets/base",
1391"_model_module_version": "1.2.0",
1392"_model_name": "LayoutModel",
1393"_view_count": null,
1394"_view_module": "@jupyter-widgets/base",
1395"_view_module_version": "1.2.0",
1396"_view_name": "LayoutView",
1397"align_content": null,
1398"align_items": null,
1399"align_self": null,
1400"border": null,
1401"bottom": null,
1402"display": null,
1403"flex": null,
1404"flex_flow": null,
1405"grid_area": null,
1406"grid_auto_columns": null,
1407"grid_auto_flow": null,
1408"grid_auto_rows": null,
1409"grid_column": null,
1410"grid_gap": null,
1411"grid_row": null,
1412"grid_template_areas": null,
1413"grid_template_columns": null,
1414"grid_template_rows": null,
1415"height": null,
1416"justify_content": null,
1417"justify_items": null,
1418"left": null,
1419"margin": null,
1420"max_height": null,
1421"max_width": null,
1422"min_height": null,
1423"min_width": null,
1424"object_fit": null,
1425"object_position": null,
1426"order": null,
1427"overflow": null,
1428"overflow_x": null,
1429"overflow_y": null,
1430"padding": null,
1431"right": null,
1432"top": null,
1433"visibility": null,
1434"width": null
1435}
1436},
1437"2ac3ffc96b1348e5997cf6d3a78720cd": {
1438"model_module": "@jupyter-widgets/base",
1439"model_module_version": "1.2.0",
1440"model_name": "LayoutModel",
1441"state": {
1442"_model_module": "@jupyter-widgets/base",
1443"_model_module_version": "1.2.0",
1444"_model_name": "LayoutModel",
1445"_view_count": null,
1446"_view_module": "@jupyter-widgets/base",
1447"_view_module_version": "1.2.0",
1448"_view_name": "LayoutView",
1449"align_content": null,
1450"align_items": null,
1451"align_self": null,
1452"border": null,
1453"bottom": null,
1454"display": null,
1455"flex": null,
1456"flex_flow": null,
1457"grid_area": null,
1458"grid_auto_columns": null,
1459"grid_auto_flow": null,
1460"grid_auto_rows": null,
1461"grid_column": null,
1462"grid_gap": null,
1463"grid_row": null,
1464"grid_template_areas": null,
1465"grid_template_columns": null,
1466"grid_template_rows": null,
1467"height": null,
1468"justify_content": null,
1469"justify_items": null,
1470"left": null,
1471"margin": null,
1472"max_height": null,
1473"max_width": null,
1474"min_height": null,
1475"min_width": null,
1476"object_fit": null,
1477"object_position": null,
1478"order": null,
1479"overflow": null,
1480"overflow_x": null,
1481"overflow_y": null,
1482"padding": null,
1483"right": null,
1484"top": null,
1485"visibility": null,
1486"width": null
1487}
1488},
1489"2dfea83397ce4cb78dc4b14d7d19f00f": {
1490"model_module": "@jupyter-widgets/controls",
1491"model_module_version": "1.5.0",
1492"model_name": "FloatProgressModel",
1493"state": {
1494"_dom_classes": [],
1495"_model_module": "@jupyter-widgets/controls",
1496"_model_module_version": "1.5.0",
1497"_model_name": "FloatProgressModel",
1498"_view_count": null,
1499"_view_module": "@jupyter-widgets/controls",
1500"_view_module_version": "1.5.0",
1501"_view_name": "ProgressView",
1502"bar_style": "",
1503"description": "",
1504"description_tooltip": null,
1505"layout": "IPY_MODEL_82536a1182ca49cb8fc103045706642f",
1506"max": 500,
1507"min": 0,
1508"orientation": "horizontal",
1509"style": "IPY_MODEL_c2c9811d883a4a619e900fd8e66bde8b",
1510"value": 500
1511}
1512},
1513"2e7a3ef5cfa34370bffe45fddb73ae8d": {
1514"model_module": "@jupyter-widgets/base",
1515"model_module_version": "1.2.0",
1516"model_name": "LayoutModel",
1517"state": {
1518"_model_module": "@jupyter-widgets/base",
1519"_model_module_version": "1.2.0",
1520"_model_name": "LayoutModel",
1521"_view_count": null,
1522"_view_module": "@jupyter-widgets/base",
1523"_view_module_version": "1.2.0",
1524"_view_name": "LayoutView",
1525"align_content": null,
1526"align_items": null,
1527"align_self": null,
1528"border": null,
1529"bottom": null,
1530"display": null,
1531"flex": null,
1532"flex_flow": null,
1533"grid_area": null,
1534"grid_auto_columns": null,
1535"grid_auto_flow": null,
1536"grid_auto_rows": null,
1537"grid_column": null,
1538"grid_gap": null,
1539"grid_row": null,
1540"grid_template_areas": null,
1541"grid_template_columns": null,
1542"grid_template_rows": null,
1543"height": null,
1544"justify_content": null,
1545"justify_items": null,
1546"left": null,
1547"margin": null,
1548"max_height": null,
1549"max_width": null,
1550"min_height": null,
1551"min_width": null,
1552"object_fit": null,
1553"object_position": null,
1554"order": null,
1555"overflow": null,
1556"overflow_x": null,
1557"overflow_y": null,
1558"padding": null,
1559"right": null,
1560"top": null,
1561"visibility": null,
1562"width": null
1563}
1564},
1565"308d349564ec408b8e18f8486ed1de85": {
1566"model_module": "@jupyter-widgets/base",
1567"model_module_version": "1.2.0",
1568"model_name": "LayoutModel",
1569"state": {
1570"_model_module": "@jupyter-widgets/base",
1571"_model_module_version": "1.2.0",
1572"_model_name": "LayoutModel",
1573"_view_count": null,
1574"_view_module": "@jupyter-widgets/base",
1575"_view_module_version": "1.2.0",
1576"_view_name": "LayoutView",
1577"align_content": null,
1578"align_items": null,
1579"align_self": null,
1580"border": null,
1581"bottom": null,
1582"display": null,
1583"flex": null,
1584"flex_flow": null,
1585"grid_area": null,
1586"grid_auto_columns": null,
1587"grid_auto_flow": null,
1588"grid_auto_rows": null,
1589"grid_column": null,
1590"grid_gap": null,
1591"grid_row": null,
1592"grid_template_areas": null,
1593"grid_template_columns": null,
1594"grid_template_rows": null,
1595"height": null,
1596"justify_content": null,
1597"justify_items": null,
1598"left": null,
1599"margin": null,
1600"max_height": null,
1601"max_width": null,
1602"min_height": null,
1603"min_width": null,
1604"object_fit": null,
1605"object_position": null,
1606"order": null,
1607"overflow": null,
1608"overflow_x": null,
1609"overflow_y": null,
1610"padding": null,
1611"right": null,
1612"top": null,
1613"visibility": null,
1614"width": null
1615}
1616},
1617"32b6df9a4f2b42739355c896cea4ca4f": {
1618"model_module": "@jupyter-widgets/base",
1619"model_module_version": "1.2.0",
1620"model_name": "LayoutModel",
1621"state": {
1622"_model_module": "@jupyter-widgets/base",
1623"_model_module_version": "1.2.0",
1624"_model_name": "LayoutModel",
1625"_view_count": null,
1626"_view_module": "@jupyter-widgets/base",
1627"_view_module_version": "1.2.0",
1628"_view_name": "LayoutView",
1629"align_content": null,
1630"align_items": null,
1631"align_self": null,
1632"border": null,
1633"bottom": null,
1634"display": null,
1635"flex": null,
1636"flex_flow": null,
1637"grid_area": null,
1638"grid_auto_columns": null,
1639"grid_auto_flow": null,
1640"grid_auto_rows": null,
1641"grid_column": null,
1642"grid_gap": null,
1643"grid_row": null,
1644"grid_template_areas": null,
1645"grid_template_columns": null,
1646"grid_template_rows": null,
1647"height": null,
1648"justify_content": null,
1649"justify_items": null,
1650"left": null,
1651"margin": null,
1652"max_height": null,
1653"max_width": null,
1654"min_height": null,
1655"min_width": null,
1656"object_fit": null,
1657"object_position": null,
1658"order": null,
1659"overflow": null,
1660"overflow_x": null,
1661"overflow_y": null,
1662"padding": null,
1663"right": null,
1664"top": null,
1665"visibility": null,
1666"width": null
1667}
1668},
1669"3415e17e7a524cdea62af74b89050530": {
1670"model_module": "@jupyter-widgets/base",
1671"model_module_version": "1.2.0",
1672"model_name": "LayoutModel",
1673"state": {
1674"_model_module": "@jupyter-widgets/base",
1675"_model_module_version": "1.2.0",
1676"_model_name": "LayoutModel",
1677"_view_count": null,
1678"_view_module": "@jupyter-widgets/base",
1679"_view_module_version": "1.2.0",
1680"_view_name": "LayoutView",
1681"align_content": null,
1682"align_items": null,
1683"align_self": null,
1684"border": null,
1685"bottom": null,
1686"display": null,
1687"flex": null,
1688"flex_flow": null,
1689"grid_area": null,
1690"grid_auto_columns": null,
1691"grid_auto_flow": null,
1692"grid_auto_rows": null,
1693"grid_column": null,
1694"grid_gap": null,
1695"grid_row": null,
1696"grid_template_areas": null,
1697"grid_template_columns": null,
1698"grid_template_rows": null,
1699"height": null,
1700"justify_content": null,
1701"justify_items": null,
1702"left": null,
1703"margin": null,
1704"max_height": null,
1705"max_width": null,
1706"min_height": null,
1707"min_width": null,
1708"object_fit": null,
1709"object_position": null,
1710"order": null,
1711"overflow": null,
1712"overflow_x": null,
1713"overflow_y": null,
1714"padding": null,
1715"right": null,
1716"top": null,
1717"visibility": null,
1718"width": null
1719}
1720},
1721"39b8642c3fbb4bc5b6f0be6202b62307": {
1722"model_module": "@jupyter-widgets/base",
1723"model_module_version": "1.2.0",
1724"model_name": "LayoutModel",
1725"state": {
1726"_model_module": "@jupyter-widgets/base",
1727"_model_module_version": "1.2.0",
1728"_model_name": "LayoutModel",
1729"_view_count": null,
1730"_view_module": "@jupyter-widgets/base",
1731"_view_module_version": "1.2.0",
1732"_view_name": "LayoutView",
1733"align_content": null,
1734"align_items": null,
1735"align_self": null,
1736"border": null,
1737"bottom": null,
1738"display": null,
1739"flex": null,
1740"flex_flow": null,
1741"grid_area": null,
1742"grid_auto_columns": null,
1743"grid_auto_flow": null,
1744"grid_auto_rows": null,
1745"grid_column": null,
1746"grid_gap": null,
1747"grid_row": null,
1748"grid_template_areas": null,
1749"grid_template_columns": null,
1750"grid_template_rows": null,
1751"height": null,
1752"justify_content": null,
1753"justify_items": null,
1754"left": null,
1755"margin": null,
1756"max_height": null,
1757"max_width": null,
1758"min_height": null,
1759"min_width": null,
1760"object_fit": null,
1761"object_position": null,
1762"order": null,
1763"overflow": null,
1764"overflow_x": null,
1765"overflow_y": null,
1766"padding": null,
1767"right": null,
1768"top": null,
1769"visibility": null,
1770"width": null
1771}
1772},
1773"3e7b0222f5024a599892a17d78d13526": {
1774"model_module": "@jupyter-widgets/base",
1775"model_module_version": "1.2.0",
1776"model_name": "LayoutModel",
1777"state": {
1778"_model_module": "@jupyter-widgets/base",
1779"_model_module_version": "1.2.0",
1780"_model_name": "LayoutModel",
1781"_view_count": null,
1782"_view_module": "@jupyter-widgets/base",
1783"_view_module_version": "1.2.0",
1784"_view_name": "LayoutView",
1785"align_content": null,
1786"align_items": null,
1787"align_self": null,
1788"border": null,
1789"bottom": null,
1790"display": null,
1791"flex": null,
1792"flex_flow": null,
1793"grid_area": null,
1794"grid_auto_columns": null,
1795"grid_auto_flow": null,
1796"grid_auto_rows": null,
1797"grid_column": null,
1798"grid_gap": null,
1799"grid_row": null,
1800"grid_template_areas": null,
1801"grid_template_columns": null,
1802"grid_template_rows": null,
1803"height": null,
1804"justify_content": null,
1805"justify_items": null,
1806"left": null,
1807"margin": null,
1808"max_height": null,
1809"max_width": null,
1810"min_height": null,
1811"min_width": null,
1812"object_fit": null,
1813"object_position": null,
1814"order": null,
1815"overflow": null,
1816"overflow_x": null,
1817"overflow_y": null,
1818"padding": null,
1819"right": null,
1820"top": null,
1821"visibility": null,
1822"width": null
1823}
1824},
1825"3ecef000cd494737a7ba938505ac55de": {
1826"model_module": "@jupyter-widgets/controls",
1827"model_module_version": "1.5.0",
1828"model_name": "HBoxModel",
1829"state": {
1830"_dom_classes": [],
1831"_model_module": "@jupyter-widgets/controls",
1832"_model_module_version": "1.5.0",
1833"_model_name": "HBoxModel",
1834"_view_count": null,
1835"_view_module": "@jupyter-widgets/controls",
1836"_view_module_version": "1.5.0",
1837"_view_name": "HBoxView",
1838"box_style": "",
1839"children": [
1840"IPY_MODEL_bd3edc0d7c174ec09a3b8a881a22cbaf",
1841"IPY_MODEL_a3b312ea3f3a4558a4e0db94b1561a64",
1842"IPY_MODEL_1ad595db69bd4f9caf30b46027bfde96"
1843],
1844"layout": "IPY_MODEL_8d9f9db2e8704eb18ca2a1920cb06ba2"
1845}
1846},
1847"40b432b310e44844b8291471c1521f20": {
1848"model_module": "@jupyter-widgets/controls",
1849"model_module_version": "1.5.0",
1850"model_name": "HTMLModel",
1851"state": {
1852"_dom_classes": [],
1853"_model_module": "@jupyter-widgets/controls",
1854"_model_module_version": "1.5.0",
1855"_model_name": "HTMLModel",
1856"_view_count": null,
1857"_view_module": "@jupyter-widgets/controls",
1858"_view_module_version": "1.5.0",
1859"_view_name": "HTMLView",
1860"description": "",
1861"description_tooltip": null,
1862"layout": "IPY_MODEL_3e7b0222f5024a599892a17d78d13526",
1863"placeholder": "",
1864"style": "IPY_MODEL_7b19a341d5174298b74b822929fe3cb9",
1865"value": " 336k/336k [00:00<00:00, 1.14MB/s]"
1866}
1867},
1868"41cf0a3151be4597bee89c3c6d3b9a27": {
1869"model_module": "@jupyter-widgets/base",
1870"model_module_version": "1.2.0",
1871"model_name": "LayoutModel",
1872"state": {
1873"_model_module": "@jupyter-widgets/base",
1874"_model_module_version": "1.2.0",
1875"_model_name": "LayoutModel",
1876"_view_count": null,
1877"_view_module": "@jupyter-widgets/base",
1878"_view_module_version": "1.2.0",
1879"_view_name": "LayoutView",
1880"align_content": null,
1881"align_items": null,
1882"align_self": null,
1883"border": null,
1884"bottom": null,
1885"display": null,
1886"flex": null,
1887"flex_flow": null,
1888"grid_area": null,
1889"grid_auto_columns": null,
1890"grid_auto_flow": null,
1891"grid_auto_rows": null,
1892"grid_column": null,
1893"grid_gap": null,
1894"grid_row": null,
1895"grid_template_areas": null,
1896"grid_template_columns": null,
1897"grid_template_rows": null,
1898"height": null,
1899"justify_content": null,
1900"justify_items": null,
1901"left": null,
1902"margin": null,
1903"max_height": null,
1904"max_width": null,
1905"min_height": null,
1906"min_width": null,
1907"object_fit": null,
1908"object_position": null,
1909"order": null,
1910"overflow": null,
1911"overflow_x": null,
1912"overflow_y": null,
1913"padding": null,
1914"right": null,
1915"top": null,
1916"visibility": null,
1917"width": null
1918}
1919},
1920"4239287597f344919e756848c5e1536a": {
1921"model_module": "@jupyter-widgets/controls",
1922"model_module_version": "1.5.0",
1923"model_name": "HTMLModel",
1924"state": {
1925"_dom_classes": [],
1926"_model_module": "@jupyter-widgets/controls",
1927"_model_module_version": "1.5.0",
1928"_model_name": "HTMLModel",
1929"_view_count": null,
1930"_view_module": "@jupyter-widgets/controls",
1931"_view_module_version": "1.5.0",
1932"_view_name": "HTMLView",
1933"description": "",
1934"description_tooltip": null,
1935"layout": "IPY_MODEL_ea02fb4d9fbe451fabffa0001ff8a1a4",
1936"placeholder": "",
1937"style": "IPY_MODEL_c7e26ccc3f574a39b80553fdb9c3b70e",
1938"value": "Extracting data files: 100%"
1939}
1940},
1941"45dab7f0fc3b4e0e89e0a3193309b834": {
1942"model_module": "@jupyter-widgets/base",
1943"model_module_version": "1.2.0",
1944"model_name": "LayoutModel",
1945"state": {
1946"_model_module": "@jupyter-widgets/base",
1947"_model_module_version": "1.2.0",
1948"_model_name": "LayoutModel",
1949"_view_count": null,
1950"_view_module": "@jupyter-widgets/base",
1951"_view_module_version": "1.2.0",
1952"_view_name": "LayoutView",
1953"align_content": null,
1954"align_items": null,
1955"align_self": null,
1956"border": null,
1957"bottom": null,
1958"display": null,
1959"flex": null,
1960"flex_flow": null,
1961"grid_area": null,
1962"grid_auto_columns": null,
1963"grid_auto_flow": null,
1964"grid_auto_rows": null,
1965"grid_column": null,
1966"grid_gap": null,
1967"grid_row": null,
1968"grid_template_areas": null,
1969"grid_template_columns": null,
1970"grid_template_rows": null,
1971"height": null,
1972"justify_content": null,
1973"justify_items": null,
1974"left": null,
1975"margin": null,
1976"max_height": null,
1977"max_width": null,
1978"min_height": null,
1979"min_width": null,
1980"object_fit": null,
1981"object_position": null,
1982"order": null,
1983"overflow": null,
1984"overflow_x": null,
1985"overflow_y": null,
1986"padding": null,
1987"right": null,
1988"top": null,
1989"visibility": null,
1990"width": null
1991}
1992},
1993"4757ba7fe9b644faad80e3910978457d": {
1994"model_module": "@jupyter-widgets/base",
1995"model_module_version": "1.2.0",
1996"model_name": "LayoutModel",
1997"state": {
1998"_model_module": "@jupyter-widgets/base",
1999"_model_module_version": "1.2.0",
2000"_model_name": "LayoutModel",
2001"_view_count": null,
2002"_view_module": "@jupyter-widgets/base",
2003"_view_module_version": "1.2.0",
2004"_view_name": "LayoutView",
2005"align_content": null,
2006"align_items": null,
2007"align_self": null,
2008"border": null,
2009"bottom": null,
2010"display": null,
2011"flex": null,
2012"flex_flow": null,
2013"grid_area": null,
2014"grid_auto_columns": null,
2015"grid_auto_flow": null,
2016"grid_auto_rows": null,
2017"grid_column": null,
2018"grid_gap": null,
2019"grid_row": null,
2020"grid_template_areas": null,
2021"grid_template_columns": null,
2022"grid_template_rows": null,
2023"height": null,
2024"justify_content": null,
2025"justify_items": null,
2026"left": null,
2027"margin": null,
2028"max_height": null,
2029"max_width": null,
2030"min_height": null,
2031"min_width": null,
2032"object_fit": null,
2033"object_position": null,
2034"order": null,
2035"overflow": null,
2036"overflow_x": null,
2037"overflow_y": null,
2038"padding": null,
2039"right": null,
2040"top": null,
2041"visibility": null,
2042"width": null
2043}
2044},
2045"4c3ed023db9b4cf1b6cbd6e763829232": {
2046"model_module": "@jupyter-widgets/controls",
2047"model_module_version": "1.5.0",
2048"model_name": "DescriptionStyleModel",
2049"state": {
2050"_model_module": "@jupyter-widgets/controls",
2051"_model_module_version": "1.5.0",
2052"_model_name": "DescriptionStyleModel",
2053"_view_count": null,
2054"_view_module": "@jupyter-widgets/base",
2055"_view_module_version": "1.2.0",
2056"_view_name": "StyleView",
2057"description_width": ""
2058}
2059},
2060"5b524ec90a634846a9de9b22171a036f": {
2061"model_module": "@jupyter-widgets/controls",
2062"model_module_version": "1.5.0",
2063"model_name": "HBoxModel",
2064"state": {
2065"_dom_classes": [],
2066"_model_module": "@jupyter-widgets/controls",
2067"_model_module_version": "1.5.0",
2068"_model_name": "HBoxModel",
2069"_view_count": null,
2070"_view_module": "@jupyter-widgets/controls",
2071"_view_module_version": "1.5.0",
2072"_view_name": "HBoxView",
2073"box_style": "",
2074"children": [
2075"IPY_MODEL_be9f9665d9b4421387d466b36dc12e53",
2076"IPY_MODEL_e999e472f30646119a333aedcf1eb42c",
2077"IPY_MODEL_40b432b310e44844b8291471c1521f20"
2078],
2079"layout": "IPY_MODEL_b99172f8069349d0ab155f77b35ee6c6"
2080}
2081},
2082"5c730e1a02104929bf61a227dfaea883": {
2083"model_module": "@jupyter-widgets/base",
2084"model_module_version": "1.2.0",
2085"model_name": "LayoutModel",
2086"state": {
2087"_model_module": "@jupyter-widgets/base",
2088"_model_module_version": "1.2.0",
2089"_model_name": "LayoutModel",
2090"_view_count": null,
2091"_view_module": "@jupyter-widgets/base",
2092"_view_module_version": "1.2.0",
2093"_view_name": "LayoutView",
2094"align_content": null,
2095"align_items": null,
2096"align_self": null,
2097"border": null,
2098"bottom": null,
2099"display": null,
2100"flex": null,
2101"flex_flow": null,
2102"grid_area": null,
2103"grid_auto_columns": null,
2104"grid_auto_flow": null,
2105"grid_auto_rows": null,
2106"grid_column": null,
2107"grid_gap": null,
2108"grid_row": null,
2109"grid_template_areas": null,
2110"grid_template_columns": null,
2111"grid_template_rows": null,
2112"height": null,
2113"justify_content": null,
2114"justify_items": null,
2115"left": null,
2116"margin": null,
2117"max_height": null,
2118"max_width": null,
2119"min_height": null,
2120"min_width": null,
2121"object_fit": null,
2122"object_position": null,
2123"order": null,
2124"overflow": null,
2125"overflow_x": null,
2126"overflow_y": null,
2127"padding": null,
2128"right": null,
2129"top": null,
2130"visibility": null,
2131"width": null
2132}
2133},
2134"62b6a3221bb745eeb13403708f0487bf": {
2135"model_module": "@jupyter-widgets/controls",
2136"model_module_version": "1.5.0",
2137"model_name": "HTMLModel",
2138"state": {
2139"_dom_classes": [],
2140"_model_module": "@jupyter-widgets/controls",
2141"_model_module_version": "1.5.0",
2142"_model_name": "HTMLModel",
2143"_view_count": null,
2144"_view_module": "@jupyter-widgets/controls",
2145"_view_module_version": "1.5.0",
2146"_view_name": "HTMLView",
2147"description": "",
2148"description_tooltip": null,
2149"layout": "IPY_MODEL_308d349564ec408b8e18f8486ed1de85",
2150"placeholder": "",
2151"style": "IPY_MODEL_4c3ed023db9b4cf1b6cbd6e763829232",
2152"value": "Downloading data files: 100%"
2153}
2154},
2155"62f17160914242c0a2ce825816be36c8": {
2156"model_module": "@jupyter-widgets/controls",
2157"model_module_version": "1.5.0",
2158"model_name": "DescriptionStyleModel",
2159"state": {
2160"_model_module": "@jupyter-widgets/controls",
2161"_model_module_version": "1.5.0",
2162"_model_name": "DescriptionStyleModel",
2163"_view_count": null,
2164"_view_module": "@jupyter-widgets/base",
2165"_view_module_version": "1.2.0",
2166"_view_name": "StyleView",
2167"description_width": ""
2168}
2169},
2170"6aa7ba4ec9464067a5816869e62b67c2": {
2171"model_module": "@jupyter-widgets/controls",
2172"model_module_version": "1.5.0",
2173"model_name": "HTMLModel",
2174"state": {
2175"_dom_classes": [],
2176"_model_module": "@jupyter-widgets/controls",
2177"_model_module_version": "1.5.0",
2178"_model_name": "HTMLModel",
2179"_view_count": null,
2180"_view_module": "@jupyter-widgets/controls",
2181"_view_module_version": "1.5.0",
2182"_view_name": "HTMLView",
2183"description": "",
2184"description_tooltip": null,
2185"layout": "IPY_MODEL_3415e17e7a524cdea62af74b89050530",
2186"placeholder": "",
2187"style": "IPY_MODEL_28a99f10e2eb42a5b8e86be49f7186dc",
2188"value": " 2.99k/? [00:00<00:00, 67.8kB/s]"
2189}
2190},
2191"6b9659cabd534ecb983fc84da293cfdc": {
2192"model_module": "@jupyter-widgets/controls",
2193"model_module_version": "1.5.0",
2194"model_name": "ProgressStyleModel",
2195"state": {
2196"_model_module": "@jupyter-widgets/controls",
2197"_model_module_version": "1.5.0",
2198"_model_name": "ProgressStyleModel",
2199"_view_count": null,
2200"_view_module": "@jupyter-widgets/base",
2201"_view_module_version": "1.2.0",
2202"_view_name": "StyleView",
2203"bar_color": null,
2204"description_width": ""
2205}
2206},
2207"6de1c75f642d484b856b36ec49d68094": {
2208"model_module": "@jupyter-widgets/base",
2209"model_module_version": "1.2.0",
2210"model_name": "LayoutModel",
2211"state": {
2212"_model_module": "@jupyter-widgets/base",
2213"_model_module_version": "1.2.0",
2214"_model_name": "LayoutModel",
2215"_view_count": null,
2216"_view_module": "@jupyter-widgets/base",
2217"_view_module_version": "1.2.0",
2218"_view_name": "LayoutView",
2219"align_content": null,
2220"align_items": null,
2221"align_self": null,
2222"border": null,
2223"bottom": null,
2224"display": null,
2225"flex": null,
2226"flex_flow": null,
2227"grid_area": null,
2228"grid_auto_columns": null,
2229"grid_auto_flow": null,
2230"grid_auto_rows": null,
2231"grid_column": null,
2232"grid_gap": null,
2233"grid_row": null,
2234"grid_template_areas": null,
2235"grid_template_columns": null,
2236"grid_template_rows": null,
2237"height": null,
2238"justify_content": null,
2239"justify_items": null,
2240"left": null,
2241"margin": null,
2242"max_height": null,
2243"max_width": null,
2244"min_height": null,
2245"min_width": null,
2246"object_fit": null,
2247"object_position": null,
2248"order": null,
2249"overflow": null,
2250"overflow_x": null,
2251"overflow_y": null,
2252"padding": null,
2253"right": null,
2254"top": null,
2255"visibility": null,
2256"width": null
2257}
2258},
2259"7b19a341d5174298b74b822929fe3cb9": {
2260"model_module": "@jupyter-widgets/controls",
2261"model_module_version": "1.5.0",
2262"model_name": "DescriptionStyleModel",
2263"state": {
2264"_model_module": "@jupyter-widgets/controls",
2265"_model_module_version": "1.5.0",
2266"_model_name": "DescriptionStyleModel",
2267"_view_count": null,
2268"_view_module": "@jupyter-widgets/base",
2269"_view_module_version": "1.2.0",
2270"_view_name": "StyleView",
2271"description_width": ""
2272}
2273},
2274"7ebfff0fd00f43a4a3c68b6955894e24": {
2275"model_module": "@jupyter-widgets/base",
2276"model_module_version": "1.2.0",
2277"model_name": "LayoutModel",
2278"state": {
2279"_model_module": "@jupyter-widgets/base",
2280"_model_module_version": "1.2.0",
2281"_model_name": "LayoutModel",
2282"_view_count": null,
2283"_view_module": "@jupyter-widgets/base",
2284"_view_module_version": "1.2.0",
2285"_view_name": "LayoutView",
2286"align_content": null,
2287"align_items": null,
2288"align_self": null,
2289"border": null,
2290"bottom": null,
2291"display": null,
2292"flex": null,
2293"flex_flow": null,
2294"grid_area": null,
2295"grid_auto_columns": null,
2296"grid_auto_flow": null,
2297"grid_auto_rows": null,
2298"grid_column": null,
2299"grid_gap": null,
2300"grid_row": null,
2301"grid_template_areas": null,
2302"grid_template_columns": null,
2303"grid_template_rows": null,
2304"height": null,
2305"justify_content": null,
2306"justify_items": null,
2307"left": null,
2308"margin": null,
2309"max_height": null,
2310"max_width": null,
2311"min_height": null,
2312"min_width": null,
2313"object_fit": null,
2314"object_position": null,
2315"order": null,
2316"overflow": null,
2317"overflow_x": null,
2318"overflow_y": null,
2319"padding": null,
2320"right": null,
2321"top": null,
2322"visibility": null,
2323"width": null
2324}
2325},
2326"82536a1182ca49cb8fc103045706642f": {
2327"model_module": "@jupyter-widgets/base",
2328"model_module_version": "1.2.0",
2329"model_name": "LayoutModel",
2330"state": {
2331"_model_module": "@jupyter-widgets/base",
2332"_model_module_version": "1.2.0",
2333"_model_name": "LayoutModel",
2334"_view_count": null,
2335"_view_module": "@jupyter-widgets/base",
2336"_view_module_version": "1.2.0",
2337"_view_name": "LayoutView",
2338"align_content": null,
2339"align_items": null,
2340"align_self": null,
2341"border": null,
2342"bottom": null,
2343"display": null,
2344"flex": null,
2345"flex_flow": null,
2346"grid_area": null,
2347"grid_auto_columns": null,
2348"grid_auto_flow": null,
2349"grid_auto_rows": null,
2350"grid_column": null,
2351"grid_gap": null,
2352"grid_row": null,
2353"grid_template_areas": null,
2354"grid_template_columns": null,
2355"grid_template_rows": null,
2356"height": null,
2357"justify_content": null,
2358"justify_items": null,
2359"left": null,
2360"margin": null,
2361"max_height": null,
2362"max_width": null,
2363"min_height": null,
2364"min_width": null,
2365"object_fit": null,
2366"object_position": null,
2367"order": null,
2368"overflow": null,
2369"overflow_x": null,
2370"overflow_y": null,
2371"padding": null,
2372"right": null,
2373"top": null,
2374"visibility": null,
2375"width": null
2376}
2377},
2378"85fd7924781a40aab82e823e8e860909": {
2379"model_module": "@jupyter-widgets/controls",
2380"model_module_version": "1.5.0",
2381"model_name": "HTMLModel",
2382"state": {
2383"_dom_classes": [],
2384"_model_module": "@jupyter-widgets/controls",
2385"_model_module_version": "1.5.0",
2386"_model_name": "HTMLModel",
2387"_view_count": null,
2388"_view_module": "@jupyter-widgets/controls",
2389"_view_module_version": "1.5.0",
2390"_view_name": "HTMLView",
2391"description": "",
2392"description_tooltip": null,
2393"layout": "IPY_MODEL_a6cad1f5f5724757b62773d3e65f9cb3",
2394"placeholder": "",
2395"style": "IPY_MODEL_8975440fb35246f5b98085fc4f78f03f",
2396"value": "Downloading metadata: "
2397}
2398},
2399"8975440fb35246f5b98085fc4f78f03f": {
2400"model_module": "@jupyter-widgets/controls",
2401"model_module_version": "1.5.0",
2402"model_name": "DescriptionStyleModel",
2403"state": {
2404"_model_module": "@jupyter-widgets/controls",
2405"_model_module_version": "1.5.0",
2406"_model_name": "DescriptionStyleModel",
2407"_view_count": null,
2408"_view_module": "@jupyter-widgets/base",
2409"_view_module_version": "1.2.0",
2410"_view_name": "StyleView",
2411"description_width": ""
2412}
2413},
2414"8a7c916f389541279a5791a4e4372cf5": {
2415"model_module": "@jupyter-widgets/controls",
2416"model_module_version": "1.5.0",
2417"model_name": "DescriptionStyleModel",
2418"state": {
2419"_model_module": "@jupyter-widgets/controls",
2420"_model_module_version": "1.5.0",
2421"_model_name": "DescriptionStyleModel",
2422"_view_count": null,
2423"_view_module": "@jupyter-widgets/base",
2424"_view_module_version": "1.2.0",
2425"_view_name": "StyleView",
2426"description_width": ""
2427}
2428},
2429"8d9f9db2e8704eb18ca2a1920cb06ba2": {
2430"model_module": "@jupyter-widgets/base",
2431"model_module_version": "1.2.0",
2432"model_name": "LayoutModel",
2433"state": {
2434"_model_module": "@jupyter-widgets/base",
2435"_model_module_version": "1.2.0",
2436"_model_name": "LayoutModel",
2437"_view_count": null,
2438"_view_module": "@jupyter-widgets/base",
2439"_view_module_version": "1.2.0",
2440"_view_name": "LayoutView",
2441"align_content": null,
2442"align_items": null,
2443"align_self": null,
2444"border": null,
2445"bottom": null,
2446"display": null,
2447"flex": null,
2448"flex_flow": null,
2449"grid_area": null,
2450"grid_auto_columns": null,
2451"grid_auto_flow": null,
2452"grid_auto_rows": null,
2453"grid_column": null,
2454"grid_gap": null,
2455"grid_row": null,
2456"grid_template_areas": null,
2457"grid_template_columns": null,
2458"grid_template_rows": null,
2459"height": null,
2460"justify_content": null,
2461"justify_items": null,
2462"left": null,
2463"margin": null,
2464"max_height": null,
2465"max_width": null,
2466"min_height": null,
2467"min_width": null,
2468"object_fit": null,
2469"object_position": null,
2470"order": null,
2471"overflow": null,
2472"overflow_x": null,
2473"overflow_y": null,
2474"padding": null,
2475"right": null,
2476"top": null,
2477"visibility": null,
2478"width": null
2479}
2480},
2481"8fe2d4de448a4703b9ce6c52a5758eb4": {
2482"model_module": "@jupyter-widgets/controls",
2483"model_module_version": "1.5.0",
2484"model_name": "FloatProgressModel",
2485"state": {
2486"_dom_classes": [],
2487"_model_module": "@jupyter-widgets/controls",
2488"_model_module_version": "1.5.0",
2489"_model_name": "FloatProgressModel",
2490"_view_count": null,
2491"_view_module": "@jupyter-widgets/controls",
2492"_view_module_version": "1.5.0",
2493"_view_name": "ProgressView",
2494"bar_style": "success",
2495"description": "",
2496"description_tooltip": null,
2497"layout": "IPY_MODEL_e4d008ff14684317a873c7607a1fec0a",
2498"max": 1416,
2499"min": 0,
2500"orientation": "horizontal",
2501"style": "IPY_MODEL_1c6cd51d6bed4f5fbb0993b957c22c84",
2502"value": 1416
2503}
2504},
2505"a27c9808eba4418aabf40a9e97356c4a": {
2506"model_module": "@jupyter-widgets/controls",
2507"model_module_version": "1.5.0",
2508"model_name": "FloatProgressModel",
2509"state": {
2510"_dom_classes": [],
2511"_model_module": "@jupyter-widgets/controls",
2512"_model_module_version": "1.5.0",
2513"_model_name": "FloatProgressModel",
2514"_view_count": null,
2515"_view_module": "@jupyter-widgets/controls",
2516"_view_module_version": "1.5.0",
2517"_view_name": "ProgressView",
2518"bar_style": "success",
2519"description": "",
2520"description_tooltip": null,
2521"layout": "IPY_MODEL_5c730e1a02104929bf61a227dfaea883",
2522"max": 2,
2523"min": 0,
2524"orientation": "horizontal",
2525"style": "IPY_MODEL_f0dbac8f3ac4449db25e33fef3840665",
2526"value": 2
2527}
2528},
2529"a3b312ea3f3a4558a4e0db94b1561a64": {
2530"model_module": "@jupyter-widgets/controls",
2531"model_module_version": "1.5.0",
2532"model_name": "FloatProgressModel",
2533"state": {
2534"_dom_classes": [],
2535"_model_module": "@jupyter-widgets/controls",
2536"_model_module_version": "1.5.0",
2537"_model_name": "FloatProgressModel",
2538"_view_count": null,
2539"_view_module": "@jupyter-widgets/controls",
2540"_view_module_version": "1.5.0",
2541"_view_name": "ProgressView",
2542"bar_style": "success",
2543"description": "",
2544"description_tooltip": null,
2545"layout": "IPY_MODEL_ee7b63007cc3482aa89857c9bce99e22",
2546"max": 23354,
2547"min": 0,
2548"orientation": "horizontal",
2549"style": "IPY_MODEL_f51393b65613425e9f732b35f8e94e24",
2550"value": 23354
2551}
2552},
2553"a57e5e6663944c478d949f2ec12ca51b": {
2554"model_module": "@jupyter-widgets/base",
2555"model_module_version": "1.2.0",
2556"model_name": "LayoutModel",
2557"state": {
2558"_model_module": "@jupyter-widgets/base",
2559"_model_module_version": "1.2.0",
2560"_model_name": "LayoutModel",
2561"_view_count": null,
2562"_view_module": "@jupyter-widgets/base",
2563"_view_module_version": "1.2.0",
2564"_view_name": "LayoutView",
2565"align_content": null,
2566"align_items": null,
2567"align_self": null,
2568"border": null,
2569"bottom": null,
2570"display": null,
2571"flex": null,
2572"flex_flow": null,
2573"grid_area": null,
2574"grid_auto_columns": null,
2575"grid_auto_flow": null,
2576"grid_auto_rows": null,
2577"grid_column": null,
2578"grid_gap": null,
2579"grid_row": null,
2580"grid_template_areas": null,
2581"grid_template_columns": null,
2582"grid_template_rows": null,
2583"height": null,
2584"justify_content": null,
2585"justify_items": null,
2586"left": null,
2587"margin": null,
2588"max_height": null,
2589"max_width": null,
2590"min_height": null,
2591"min_width": null,
2592"object_fit": null,
2593"object_position": null,
2594"order": null,
2595"overflow": null,
2596"overflow_x": null,
2597"overflow_y": null,
2598"padding": null,
2599"right": null,
2600"top": null,
2601"visibility": null,
2602"width": null
2603}
2604},
2605"a6cad1f5f5724757b62773d3e65f9cb3": {
2606"model_module": "@jupyter-widgets/base",
2607"model_module_version": "1.2.0",
2608"model_name": "LayoutModel",
2609"state": {
2610"_model_module": "@jupyter-widgets/base",
2611"_model_module_version": "1.2.0",
2612"_model_name": "LayoutModel",
2613"_view_count": null,
2614"_view_module": "@jupyter-widgets/base",
2615"_view_module_version": "1.2.0",
2616"_view_name": "LayoutView",
2617"align_content": null,
2618"align_items": null,
2619"align_self": null,
2620"border": null,
2621"bottom": null,
2622"display": null,
2623"flex": null,
2624"flex_flow": null,
2625"grid_area": null,
2626"grid_auto_columns": null,
2627"grid_auto_flow": null,
2628"grid_auto_rows": null,
2629"grid_column": null,
2630"grid_gap": null,
2631"grid_row": null,
2632"grid_template_areas": null,
2633"grid_template_columns": null,
2634"grid_template_rows": null,
2635"height": null,
2636"justify_content": null,
2637"justify_items": null,
2638"left": null,
2639"margin": null,
2640"max_height": null,
2641"max_width": null,
2642"min_height": null,
2643"min_width": null,
2644"object_fit": null,
2645"object_position": null,
2646"order": null,
2647"overflow": null,
2648"overflow_x": null,
2649"overflow_y": null,
2650"padding": null,
2651"right": null,
2652"top": null,
2653"visibility": null,
2654"width": null
2655}
2656},
2657"ac50c9ba79be4493a12213e01c140289": {
2658"model_module": "@jupyter-widgets/controls",
2659"model_module_version": "1.5.0",
2660"model_name": "HTMLModel",
2661"state": {
2662"_dom_classes": [],
2663"_model_module": "@jupyter-widgets/controls",
2664"_model_module_version": "1.5.0",
2665"_model_name": "HTMLModel",
2666"_view_count": null,
2667"_view_module": "@jupyter-widgets/controls",
2668"_view_module_version": "1.5.0",
2669"_view_name": "HTMLView",
2670"description": "",
2671"description_tooltip": null,
2672"layout": "IPY_MODEL_0a93329af24246f4b77acf0e5574b335",
2673"placeholder": "",
2674"style": "IPY_MODEL_1f6e8e2374bf4b3d99727eb9ed157cb6",
2675"value": " 2/2 [00:00<00:00, 25.99it/s]"
2676}
2677},
2678"ac6de5de78b24d45a17fdb5c800683b9": {
2679"model_module": "@jupyter-widgets/controls",
2680"model_module_version": "1.5.0",
2681"model_name": "HTMLModel",
2682"state": {
2683"_dom_classes": [],
2684"_model_module": "@jupyter-widgets/controls",
2685"_model_module_version": "1.5.0",
2686"_model_name": "HTMLModel",
2687"_view_count": null,
2688"_view_module": "@jupyter-widgets/controls",
2689"_view_module_version": "1.5.0",
2690"_view_name": "HTMLView",
2691"description": "",
2692"description_tooltip": null,
2693"layout": "IPY_MODEL_028f3dc9c8d148db99ec5df93b3c8773",
2694"placeholder": "",
2695"style": "IPY_MODEL_8a7c916f389541279a5791a4e4372cf5",
2696"value": " 2/2 [00:01<00:00, 1.61it/s]"
2697}
2698},
2699"adcaf0a904a54127b212ca3751f9f95b": {
2700"model_module": "@jupyter-widgets/base",
2701"model_module_version": "1.2.0",
2702"model_name": "LayoutModel",
2703"state": {
2704"_model_module": "@jupyter-widgets/base",
2705"_model_module_version": "1.2.0",
2706"_model_name": "LayoutModel",
2707"_view_count": null,
2708"_view_module": "@jupyter-widgets/base",
2709"_view_module_version": "1.2.0",
2710"_view_name": "LayoutView",
2711"align_content": null,
2712"align_items": null,
2713"align_self": null,
2714"border": null,
2715"bottom": null,
2716"display": null,
2717"flex": null,
2718"flex_flow": null,
2719"grid_area": null,
2720"grid_auto_columns": null,
2721"grid_auto_flow": null,
2722"grid_auto_rows": null,
2723"grid_column": null,
2724"grid_gap": null,
2725"grid_row": null,
2726"grid_template_areas": null,
2727"grid_template_columns": null,
2728"grid_template_rows": null,
2729"height": null,
2730"justify_content": null,
2731"justify_items": null,
2732"left": null,
2733"margin": null,
2734"max_height": null,
2735"max_width": null,
2736"min_height": null,
2737"min_width": null,
2738"object_fit": null,
2739"object_position": null,
2740"order": null,
2741"overflow": null,
2742"overflow_x": null,
2743"overflow_y": null,
2744"padding": null,
2745"right": null,
2746"top": null,
2747"visibility": null,
2748"width": null
2749}
2750},
2751"b22413d2b4444523b39a3d52acaa8587": {
2752"model_module": "@jupyter-widgets/controls",
2753"model_module_version": "1.5.0",
2754"model_name": "HBoxModel",
2755"state": {
2756"_dom_classes": [],
2757"_model_module": "@jupyter-widgets/controls",
2758"_model_module_version": "1.5.0",
2759"_model_name": "HBoxModel",
2760"_view_count": null,
2761"_view_module": "@jupyter-widgets/controls",
2762"_view_module_version": "1.5.0",
2763"_view_name": "HBoxView",
2764"box_style": "",
2765"children": [
2766"IPY_MODEL_1316d3504b9040c580aba3eb0a7d5e0e",
2767"IPY_MODEL_c821ecea97044016a2aaa9286a672a32",
2768"IPY_MODEL_029cb2f03eda4b13b800a1db177a3f60"
2769],
2770"layout": "IPY_MODEL_39b8642c3fbb4bc5b6f0be6202b62307"
2771}
2772},
2773"b99172f8069349d0ab155f77b35ee6c6": {
2774"model_module": "@jupyter-widgets/base",
2775"model_module_version": "1.2.0",
2776"model_name": "LayoutModel",
2777"state": {
2778"_model_module": "@jupyter-widgets/base",
2779"_model_module_version": "1.2.0",
2780"_model_name": "LayoutModel",
2781"_view_count": null,
2782"_view_module": "@jupyter-widgets/base",
2783"_view_module_version": "1.2.0",
2784"_view_name": "LayoutView",
2785"align_content": null,
2786"align_items": null,
2787"align_self": null,
2788"border": null,
2789"bottom": null,
2790"display": null,
2791"flex": null,
2792"flex_flow": null,
2793"grid_area": null,
2794"grid_auto_columns": null,
2795"grid_auto_flow": null,
2796"grid_auto_rows": null,
2797"grid_column": null,
2798"grid_gap": null,
2799"grid_row": null,
2800"grid_template_areas": null,
2801"grid_template_columns": null,
2802"grid_template_rows": null,
2803"height": null,
2804"justify_content": null,
2805"justify_items": null,
2806"left": null,
2807"margin": null,
2808"max_height": null,
2809"max_width": null,
2810"min_height": null,
2811"min_width": null,
2812"object_fit": null,
2813"object_position": null,
2814"order": null,
2815"overflow": null,
2816"overflow_x": null,
2817"overflow_y": null,
2818"padding": null,
2819"right": null,
2820"top": null,
2821"visibility": null,
2822"width": null
2823}
2824},
2825"bb0b95d8f00a4f1494df44bf0d2f7a26": {
2826"model_module": "@jupyter-widgets/controls",
2827"model_module_version": "1.5.0",
2828"model_name": "HBoxModel",
2829"state": {
2830"_dom_classes": [],
2831"_model_module": "@jupyter-widgets/controls",
2832"_model_module_version": "1.5.0",
2833"_model_name": "HBoxModel",
2834"_view_count": null,
2835"_view_module": "@jupyter-widgets/controls",
2836"_view_module_version": "1.5.0",
2837"_view_name": "HBoxView",
2838"box_style": "",
2839"children": [
2840"IPY_MODEL_62b6a3221bb745eeb13403708f0487bf",
2841"IPY_MODEL_a27c9808eba4418aabf40a9e97356c4a",
2842"IPY_MODEL_ac6de5de78b24d45a17fdb5c800683b9"
2843],
2844"layout": "IPY_MODEL_bfebdd15ffff4e079e4939b9444ce467"
2845}
2846},
2847"bd3edc0d7c174ec09a3b8a881a22cbaf": {
2848"model_module": "@jupyter-widgets/controls",
2849"model_module_version": "1.5.0",
2850"model_name": "HTMLModel",
2851"state": {
2852"_dom_classes": [],
2853"_model_module": "@jupyter-widgets/controls",
2854"_model_module_version": "1.5.0",
2855"_model_name": "HTMLModel",
2856"_view_count": null,
2857"_view_module": "@jupyter-widgets/controls",
2858"_view_module_version": "1.5.0",
2859"_view_name": "HTMLView",
2860"description": "",
2861"description_tooltip": null,
2862"layout": "IPY_MODEL_45dab7f0fc3b4e0e89e0a3193309b834",
2863"placeholder": "",
2864"style": "IPY_MODEL_d27c0648c05f40ef9e175d27135ec751",
2865"value": "Downloading data: 100%"
2866}
2867},
2868"be9f9665d9b4421387d466b36dc12e53": {
2869"model_module": "@jupyter-widgets/controls",
2870"model_module_version": "1.5.0",
2871"model_name": "HTMLModel",
2872"state": {
2873"_dom_classes": [],
2874"_model_module": "@jupyter-widgets/controls",
2875"_model_module_version": "1.5.0",
2876"_model_name": "HTMLModel",
2877"_view_count": null,
2878"_view_module": "@jupyter-widgets/controls",
2879"_view_module_version": "1.5.0",
2880"_view_name": "HTMLView",
2881"description": "",
2882"description_tooltip": null,
2883"layout": "IPY_MODEL_a57e5e6663944c478d949f2ec12ca51b",
2884"placeholder": "",
2885"style": "IPY_MODEL_0e9854e5ce154135b0aee7e5ea795857",
2886"value": "Downloading data: 100%"
2887}
2888},
2889"bf415d781ee84e2fb940b64d2d0490b0": {
2890"model_module": "@jupyter-widgets/base",
2891"model_module_version": "1.2.0",
2892"model_name": "LayoutModel",
2893"state": {
2894"_model_module": "@jupyter-widgets/base",
2895"_model_module_version": "1.2.0",
2896"_model_name": "LayoutModel",
2897"_view_count": null,
2898"_view_module": "@jupyter-widgets/base",
2899"_view_module_version": "1.2.0",
2900"_view_name": "LayoutView",
2901"align_content": null,
2902"align_items": null,
2903"align_self": null,
2904"border": null,
2905"bottom": null,
2906"display": null,
2907"flex": null,
2908"flex_flow": null,
2909"grid_area": null,
2910"grid_auto_columns": null,
2911"grid_auto_flow": null,
2912"grid_auto_rows": null,
2913"grid_column": null,
2914"grid_gap": null,
2915"grid_row": null,
2916"grid_template_areas": null,
2917"grid_template_columns": null,
2918"grid_template_rows": null,
2919"height": null,
2920"justify_content": null,
2921"justify_items": null,
2922"left": null,
2923"margin": null,
2924"max_height": null,
2925"max_width": null,
2926"min_height": null,
2927"min_width": null,
2928"object_fit": null,
2929"object_position": null,
2930"order": null,
2931"overflow": null,
2932"overflow_x": null,
2933"overflow_y": null,
2934"padding": null,
2935"right": null,
2936"top": null,
2937"visibility": null,
2938"width": null
2939}
2940},
2941"bfebdd15ffff4e079e4939b9444ce467": {
2942"model_module": "@jupyter-widgets/base",
2943"model_module_version": "1.2.0",
2944"model_name": "LayoutModel",
2945"state": {
2946"_model_module": "@jupyter-widgets/base",
2947"_model_module_version": "1.2.0",
2948"_model_name": "LayoutModel",
2949"_view_count": null,
2950"_view_module": "@jupyter-widgets/base",
2951"_view_module_version": "1.2.0",
2952"_view_name": "LayoutView",
2953"align_content": null,
2954"align_items": null,
2955"align_self": null,
2956"border": null,
2957"bottom": null,
2958"display": null,
2959"flex": null,
2960"flex_flow": null,
2961"grid_area": null,
2962"grid_auto_columns": null,
2963"grid_auto_flow": null,
2964"grid_auto_rows": null,
2965"grid_column": null,
2966"grid_gap": null,
2967"grid_row": null,
2968"grid_template_areas": null,
2969"grid_template_columns": null,
2970"grid_template_rows": null,
2971"height": null,
2972"justify_content": null,
2973"justify_items": null,
2974"left": null,
2975"margin": null,
2976"max_height": null,
2977"max_width": null,
2978"min_height": null,
2979"min_width": null,
2980"object_fit": null,
2981"object_position": null,
2982"order": null,
2983"overflow": null,
2984"overflow_x": null,
2985"overflow_y": null,
2986"padding": null,
2987"right": null,
2988"top": null,
2989"visibility": null,
2990"width": null
2991}
2992},
2993"c2b5da2a0bf64686b1e9c3784bc7ba48": {
2994"model_module": "@jupyter-widgets/base",
2995"model_module_version": "1.2.0",
2996"model_name": "LayoutModel",
2997"state": {
2998"_model_module": "@jupyter-widgets/base",
2999"_model_module_version": "1.2.0",
3000"_model_name": "LayoutModel",
3001"_view_count": null,
3002"_view_module": "@jupyter-widgets/base",
3003"_view_module_version": "1.2.0",
3004"_view_name": "LayoutView",
3005"align_content": null,
3006"align_items": null,
3007"align_self": null,
3008"border": null,
3009"bottom": null,
3010"display": null,
3011"flex": null,
3012"flex_flow": null,
3013"grid_area": null,
3014"grid_auto_columns": null,
3015"grid_auto_flow": null,
3016"grid_auto_rows": null,
3017"grid_column": null,
3018"grid_gap": null,
3019"grid_row": null,
3020"grid_template_areas": null,
3021"grid_template_columns": null,
3022"grid_template_rows": null,
3023"height": null,
3024"justify_content": null,
3025"justify_items": null,
3026"left": null,
3027"margin": null,
3028"max_height": null,
3029"max_width": null,
3030"min_height": null,
3031"min_width": null,
3032"object_fit": null,
3033"object_position": null,
3034"order": null,
3035"overflow": null,
3036"overflow_x": null,
3037"overflow_y": null,
3038"padding": null,
3039"right": null,
3040"top": null,
3041"visibility": null,
3042"width": null
3043}
3044},
3045"c2c9811d883a4a619e900fd8e66bde8b": {
3046"model_module": "@jupyter-widgets/controls",
3047"model_module_version": "1.5.0",
3048"model_name": "ProgressStyleModel",
3049"state": {
3050"_model_module": "@jupyter-widgets/controls",
3051"_model_module_version": "1.5.0",
3052"_model_name": "ProgressStyleModel",
3053"_view_count": null,
3054"_view_module": "@jupyter-widgets/base",
3055"_view_module_version": "1.2.0",
3056"_view_name": "StyleView",
3057"bar_color": null,
3058"description_width": ""
3059}
3060},
3061"c70aaeea77db4db6903c5873817300b0": {
3062"model_module": "@jupyter-widgets/controls",
3063"model_module_version": "1.5.0",
3064"model_name": "ProgressStyleModel",
3065"state": {
3066"_model_module": "@jupyter-widgets/controls",
3067"_model_module_version": "1.5.0",
3068"_model_name": "ProgressStyleModel",
3069"_view_count": null,
3070"_view_module": "@jupyter-widgets/base",
3071"_view_module_version": "1.2.0",
3072"_view_name": "StyleView",
3073"bar_color": null,
3074"description_width": ""
3075}
3076},
3077"c7e26ccc3f574a39b80553fdb9c3b70e": {
3078"model_module": "@jupyter-widgets/controls",
3079"model_module_version": "1.5.0",
3080"model_name": "DescriptionStyleModel",
3081"state": {
3082"_model_module": "@jupyter-widgets/controls",
3083"_model_module_version": "1.5.0",
3084"_model_name": "DescriptionStyleModel",
3085"_view_count": null,
3086"_view_module": "@jupyter-widgets/base",
3087"_view_module_version": "1.2.0",
3088"_view_name": "StyleView",
3089"description_width": ""
3090}
3091},
3092"c7f29593c65d4df8b60e56374beebf02": {
3093"model_module": "@jupyter-widgets/controls",
3094"model_module_version": "1.5.0",
3095"model_name": "DescriptionStyleModel",
3096"state": {
3097"_model_module": "@jupyter-widgets/controls",
3098"_model_module_version": "1.5.0",
3099"_model_name": "DescriptionStyleModel",
3100"_view_count": null,
3101"_view_module": "@jupyter-widgets/base",
3102"_view_module_version": "1.2.0",
3103"_view_name": "StyleView",
3104"description_width": ""
3105}
3106},
3107"c821ecea97044016a2aaa9286a672a32": {
3108"model_module": "@jupyter-widgets/controls",
3109"model_module_version": "1.5.0",
3110"model_name": "FloatProgressModel",
3111"state": {
3112"_dom_classes": [],
3113"_model_module": "@jupyter-widgets/controls",
3114"_model_module_version": "1.5.0",
3115"_model_name": "FloatProgressModel",
3116"_view_count": null,
3117"_view_module": "@jupyter-widgets/controls",
3118"_view_module_version": "1.5.0",
3119"_view_name": "ProgressView",
3120"bar_style": "success",
3121"description": "",
3122"description_tooltip": null,
3123"layout": "IPY_MODEL_c91efde0092348888ebd08afb3dce9c3",
3124"max": 2224,
3125"min": 0,
3126"orientation": "horizontal",
3127"style": "IPY_MODEL_dbe7c2c262424b6ea04621a6b2e5281c",
3128"value": 2224
3129}
3130},
3131"c91efde0092348888ebd08afb3dce9c3": {
3132"model_module": "@jupyter-widgets/base",
3133"model_module_version": "1.2.0",
3134"model_name": "LayoutModel",
3135"state": {
3136"_model_module": "@jupyter-widgets/base",
3137"_model_module_version": "1.2.0",
3138"_model_name": "LayoutModel",
3139"_view_count": null,
3140"_view_module": "@jupyter-widgets/base",
3141"_view_module_version": "1.2.0",
3142"_view_name": "LayoutView",
3143"align_content": null,
3144"align_items": null,
3145"align_self": null,
3146"border": null,
3147"bottom": null,
3148"display": null,
3149"flex": null,
3150"flex_flow": null,
3151"grid_area": null,
3152"grid_auto_columns": null,
3153"grid_auto_flow": null,
3154"grid_auto_rows": null,
3155"grid_column": null,
3156"grid_gap": null,
3157"grid_row": null,
3158"grid_template_areas": null,
3159"grid_template_columns": null,
3160"grid_template_rows": null,
3161"height": null,
3162"justify_content": null,
3163"justify_items": null,
3164"left": null,
3165"margin": null,
3166"max_height": null,
3167"max_width": null,
3168"min_height": null,
3169"min_width": null,
3170"object_fit": null,
3171"object_position": null,
3172"order": null,
3173"overflow": null,
3174"overflow_x": null,
3175"overflow_y": null,
3176"padding": null,
3177"right": null,
3178"top": null,
3179"visibility": null,
3180"width": null
3181}
3182},
3183"d27c0648c05f40ef9e175d27135ec751": {
3184"model_module": "@jupyter-widgets/controls",
3185"model_module_version": "1.5.0",
3186"model_name": "DescriptionStyleModel",
3187"state": {
3188"_model_module": "@jupyter-widgets/controls",
3189"_model_module_version": "1.5.0",
3190"_model_name": "DescriptionStyleModel",
3191"_view_count": null,
3192"_view_module": "@jupyter-widgets/base",
3193"_view_module_version": "1.2.0",
3194"_view_name": "StyleView",
3195"description_width": ""
3196}
3197},
3198"d4da7a1fc98f44d1bae157d3c9c07942": {
3199"model_module": "@jupyter-widgets/base",
3200"model_module_version": "1.2.0",
3201"model_name": "LayoutModel",
3202"state": {
3203"_model_module": "@jupyter-widgets/base",
3204"_model_module_version": "1.2.0",
3205"_model_name": "LayoutModel",
3206"_view_count": null,
3207"_view_module": "@jupyter-widgets/base",
3208"_view_module_version": "1.2.0",
3209"_view_name": "LayoutView",
3210"align_content": null,
3211"align_items": null,
3212"align_self": null,
3213"border": null,
3214"bottom": null,
3215"display": null,
3216"flex": null,
3217"flex_flow": null,
3218"grid_area": null,
3219"grid_auto_columns": null,
3220"grid_auto_flow": null,
3221"grid_auto_rows": null,
3222"grid_column": null,
3223"grid_gap": null,
3224"grid_row": null,
3225"grid_template_areas": null,
3226"grid_template_columns": null,
3227"grid_template_rows": null,
3228"height": null,
3229"justify_content": null,
3230"justify_items": null,
3231"left": null,
3232"margin": null,
3233"max_height": null,
3234"max_width": null,
3235"min_height": null,
3236"min_width": null,
3237"object_fit": null,
3238"object_position": null,
3239"order": null,
3240"overflow": null,
3241"overflow_x": null,
3242"overflow_y": null,
3243"padding": null,
3244"right": null,
3245"top": null,
3246"visibility": null,
3247"width": null
3248}
3249},
3250"d87354f9ad594885903943925d186ae0": {
3251"model_module": "@jupyter-widgets/controls",
3252"model_module_version": "1.5.0",
3253"model_name": "HBoxModel",
3254"state": {
3255"_dom_classes": [],
3256"_model_module": "@jupyter-widgets/controls",
3257"_model_module_version": "1.5.0",
3258"_model_name": "HBoxModel",
3259"_view_count": null,
3260"_view_module": "@jupyter-widgets/controls",
3261"_view_module_version": "1.5.0",
3262"_view_name": "HBoxView",
3263"box_style": "",
3264"children": [
3265"IPY_MODEL_21e580ef7de94f068a12c9bd4cce14d6",
3266"IPY_MODEL_2dfea83397ce4cb78dc4b14d7d19f00f",
3267"IPY_MODEL_dc8bbd3f9ae44ccbb53d5ccf33254f14"
3268],
3269"layout": "IPY_MODEL_c2b5da2a0bf64686b1e9c3784bc7ba48"
3270}
3271},
3272"d959e84d5a9b40a998ce16bc4dc92fbd": {
3273"model_module": "@jupyter-widgets/controls",
3274"model_module_version": "1.5.0",
3275"model_name": "FloatProgressModel",
3276"state": {
3277"_dom_classes": [],
3278"_model_module": "@jupyter-widgets/controls",
3279"_model_module_version": "1.5.0",
3280"_model_name": "FloatProgressModel",
3281"_view_count": null,
3282"_view_module": "@jupyter-widgets/controls",
3283"_view_module_version": "1.5.0",
3284"_view_name": "ProgressView",
3285"bar_style": "success",
3286"description": "",
3287"description_tooltip": null,
3288"layout": "IPY_MODEL_adcaf0a904a54127b212ca3751f9f95b",
3289"max": 2,
3290"min": 0,
3291"orientation": "horizontal",
3292"style": "IPY_MODEL_261c85699616441783c044118d9e165f",
3293"value": 2
3294}
3295},
3296"dbe7c2c262424b6ea04621a6b2e5281c": {
3297"model_module": "@jupyter-widgets/controls",
3298"model_module_version": "1.5.0",
3299"model_name": "ProgressStyleModel",
3300"state": {
3301"_model_module": "@jupyter-widgets/controls",
3302"_model_module_version": "1.5.0",
3303"_model_name": "ProgressStyleModel",
3304"_view_count": null,
3305"_view_module": "@jupyter-widgets/base",
3306"_view_module_version": "1.2.0",
3307"_view_name": "StyleView",
3308"bar_color": null,
3309"description_width": ""
3310}
3311},
3312"dc8bbd3f9ae44ccbb53d5ccf33254f14": {
3313"model_module": "@jupyter-widgets/controls",
3314"model_module_version": "1.5.0",
3315"model_name": "HTMLModel",
3316"state": {
3317"_dom_classes": [],
3318"_model_module": "@jupyter-widgets/controls",
3319"_model_module_version": "1.5.0",
3320"_model_name": "HTMLModel",
3321"_view_count": null,
3322"_view_module": "@jupyter-widgets/controls",
3323"_view_module_version": "1.5.0",
3324"_view_name": "HTMLView",
3325"description": "",
3326"description_tooltip": null,
3327"layout": "IPY_MODEL_7ebfff0fd00f43a4a3c68b6955894e24",
3328"placeholder": "",
3329"style": "IPY_MODEL_114dcbb9bdf3404db0518ee34e9ab7fc",
3330"value": " 0/500 [00:00<?, ? examples/s]"
3331}
3332},
3333"e4d008ff14684317a873c7607a1fec0a": {
3334"model_module": "@jupyter-widgets/base",
3335"model_module_version": "1.2.0",
3336"model_name": "LayoutModel",
3337"state": {
3338"_model_module": "@jupyter-widgets/base",
3339"_model_module_version": "1.2.0",
3340"_model_name": "LayoutModel",
3341"_view_count": null,
3342"_view_module": "@jupyter-widgets/base",
3343"_view_module_version": "1.2.0",
3344"_view_name": "LayoutView",
3345"align_content": null,
3346"align_items": null,
3347"align_self": null,
3348"border": null,
3349"bottom": null,
3350"display": null,
3351"flex": null,
3352"flex_flow": null,
3353"grid_area": null,
3354"grid_auto_columns": null,
3355"grid_auto_flow": null,
3356"grid_auto_rows": null,
3357"grid_column": null,
3358"grid_gap": null,
3359"grid_row": null,
3360"grid_template_areas": null,
3361"grid_template_columns": null,
3362"grid_template_rows": null,
3363"height": null,
3364"justify_content": null,
3365"justify_items": null,
3366"left": null,
3367"margin": null,
3368"max_height": null,
3369"max_width": null,
3370"min_height": null,
3371"min_width": null,
3372"object_fit": null,
3373"object_position": null,
3374"order": null,
3375"overflow": null,
3376"overflow_x": null,
3377"overflow_y": null,
3378"padding": null,
3379"right": null,
3380"top": null,
3381"visibility": null,
3382"width": null
3383}
3384},
3385"e532f8f1a6cf433fa22410a6876f4707": {
3386"model_module": "@jupyter-widgets/controls",
3387"model_module_version": "1.5.0",
3388"model_name": "DescriptionStyleModel",
3389"state": {
3390"_model_module": "@jupyter-widgets/controls",
3391"_model_module_version": "1.5.0",
3392"_model_name": "DescriptionStyleModel",
3393"_view_count": null,
3394"_view_module": "@jupyter-widgets/base",
3395"_view_module_version": "1.2.0",
3396"_view_name": "StyleView",
3397"description_width": ""
3398}
3399},
3400"e999e472f30646119a333aedcf1eb42c": {
3401"model_module": "@jupyter-widgets/controls",
3402"model_module_version": "1.5.0",
3403"model_name": "FloatProgressModel",
3404"state": {
3405"_dom_classes": [],
3406"_model_module": "@jupyter-widgets/controls",
3407"_model_module_version": "1.5.0",
3408"_model_name": "FloatProgressModel",
3409"_view_count": null,
3410"_view_module": "@jupyter-widgets/controls",
3411"_view_module_version": "1.5.0",
3412"_view_name": "ProgressView",
3413"bar_style": "success",
3414"description": "",
3415"description_tooltip": null,
3416"layout": "IPY_MODEL_d4da7a1fc98f44d1bae157d3c9c07942",
3417"max": 335858,
3418"min": 0,
3419"orientation": "horizontal",
3420"style": "IPY_MODEL_c70aaeea77db4db6903c5873817300b0",
3421"value": 335858
3422}
3423},
3424"ea02fb4d9fbe451fabffa0001ff8a1a4": {
3425"model_module": "@jupyter-widgets/base",
3426"model_module_version": "1.2.0",
3427"model_name": "LayoutModel",
3428"state": {
3429"_model_module": "@jupyter-widgets/base",
3430"_model_module_version": "1.2.0",
3431"_model_name": "LayoutModel",
3432"_view_count": null,
3433"_view_module": "@jupyter-widgets/base",
3434"_view_module_version": "1.2.0",
3435"_view_name": "LayoutView",
3436"align_content": null,
3437"align_items": null,
3438"align_self": null,
3439"border": null,
3440"bottom": null,
3441"display": null,
3442"flex": null,
3443"flex_flow": null,
3444"grid_area": null,
3445"grid_auto_columns": null,
3446"grid_auto_flow": null,
3447"grid_auto_rows": null,
3448"grid_column": null,
3449"grid_gap": null,
3450"grid_row": null,
3451"grid_template_areas": null,
3452"grid_template_columns": null,
3453"grid_template_rows": null,
3454"height": null,
3455"justify_content": null,
3456"justify_items": null,
3457"left": null,
3458"margin": null,
3459"max_height": null,
3460"max_width": null,
3461"min_height": null,
3462"min_width": null,
3463"object_fit": null,
3464"object_position": null,
3465"order": null,
3466"overflow": null,
3467"overflow_x": null,
3468"overflow_y": null,
3469"padding": null,
3470"right": null,
3471"top": null,
3472"visibility": null,
3473"width": null
3474}
3475},
3476"ea5454b5334b40bea6d179a675d6578e": {
3477"model_module": "@jupyter-widgets/controls",
3478"model_module_version": "1.5.0",
3479"model_name": "HBoxModel",
3480"state": {
3481"_dom_classes": [],
3482"_model_module": "@jupyter-widgets/controls",
3483"_model_module_version": "1.5.0",
3484"_model_name": "HBoxModel",
3485"_view_count": null,
3486"_view_module": "@jupyter-widgets/controls",
3487"_view_module_version": "1.5.0",
3488"_view_name": "HBoxView",
3489"box_style": "",
3490"children": [
3491"IPY_MODEL_f6b507d2bdb74ebaadbbd8a9b51290d6",
3492"IPY_MODEL_f9e79559b8e34d55b669413423c3c2ae",
3493"IPY_MODEL_13d8f25b01ad4f9fb12203ee85f26d50"
3494],
3495"layout": "IPY_MODEL_2e7a3ef5cfa34370bffe45fddb73ae8d"
3496}
3497},
3498"ee7b63007cc3482aa89857c9bce99e22": {
3499"model_module": "@jupyter-widgets/base",
3500"model_module_version": "1.2.0",
3501"model_name": "LayoutModel",
3502"state": {
3503"_model_module": "@jupyter-widgets/base",
3504"_model_module_version": "1.2.0",
3505"_model_name": "LayoutModel",
3506"_view_count": null,
3507"_view_module": "@jupyter-widgets/base",
3508"_view_module_version": "1.2.0",
3509"_view_name": "LayoutView",
3510"align_content": null,
3511"align_items": null,
3512"align_self": null,
3513"border": null,
3514"bottom": null,
3515"display": null,
3516"flex": null,
3517"flex_flow": null,
3518"grid_area": null,
3519"grid_auto_columns": null,
3520"grid_auto_flow": null,
3521"grid_auto_rows": null,
3522"grid_column": null,
3523"grid_gap": null,
3524"grid_row": null,
3525"grid_template_areas": null,
3526"grid_template_columns": null,
3527"grid_template_rows": null,
3528"height": null,
3529"justify_content": null,
3530"justify_items": null,
3531"left": null,
3532"margin": null,
3533"max_height": null,
3534"max_width": null,
3535"min_height": null,
3536"min_width": null,
3537"object_fit": null,
3538"object_position": null,
3539"order": null,
3540"overflow": null,
3541"overflow_x": null,
3542"overflow_y": null,
3543"padding": null,
3544"right": null,
3545"top": null,
3546"visibility": null,
3547"width": null
3548}
3549},
3550"f0dbac8f3ac4449db25e33fef3840665": {
3551"model_module": "@jupyter-widgets/controls",
3552"model_module_version": "1.5.0",
3553"model_name": "ProgressStyleModel",
3554"state": {
3555"_model_module": "@jupyter-widgets/controls",
3556"_model_module_version": "1.5.0",
3557"_model_name": "ProgressStyleModel",
3558"_view_count": null,
3559"_view_module": "@jupyter-widgets/base",
3560"_view_module_version": "1.2.0",
3561"_view_name": "StyleView",
3562"bar_color": null,
3563"description_width": ""
3564}
3565},
3566"f2d8146acd2a43298266cf4649ef8a68": {
3567"model_module": "@jupyter-widgets/controls",
3568"model_module_version": "1.5.0",
3569"model_name": "DescriptionStyleModel",
3570"state": {
3571"_model_module": "@jupyter-widgets/controls",
3572"_model_module_version": "1.5.0",
3573"_model_name": "DescriptionStyleModel",
3574"_view_count": null,
3575"_view_module": "@jupyter-widgets/base",
3576"_view_module_version": "1.2.0",
3577"_view_name": "StyleView",
3578"description_width": ""
3579}
3580},
3581"f51393b65613425e9f732b35f8e94e24": {
3582"model_module": "@jupyter-widgets/controls",
3583"model_module_version": "1.5.0",
3584"model_name": "ProgressStyleModel",
3585"state": {
3586"_model_module": "@jupyter-widgets/controls",
3587"_model_module_version": "1.5.0",
3588"_model_name": "ProgressStyleModel",
3589"_view_count": null,
3590"_view_module": "@jupyter-widgets/base",
3591"_view_module_version": "1.2.0",
3592"_view_name": "StyleView",
3593"bar_color": null,
3594"description_width": ""
3595}
3596},
3597"f5632381613a400f88d61d520f7f27e7": {
3598"model_module": "@jupyter-widgets/base",
3599"model_module_version": "1.2.0",
3600"model_name": "LayoutModel",
3601"state": {
3602"_model_module": "@jupyter-widgets/base",
3603"_model_module_version": "1.2.0",
3604"_model_name": "LayoutModel",
3605"_view_count": null,
3606"_view_module": "@jupyter-widgets/base",
3607"_view_module_version": "1.2.0",
3608"_view_name": "LayoutView",
3609"align_content": null,
3610"align_items": null,
3611"align_self": null,
3612"border": null,
3613"bottom": null,
3614"display": null,
3615"flex": null,
3616"flex_flow": null,
3617"grid_area": null,
3618"grid_auto_columns": null,
3619"grid_auto_flow": null,
3620"grid_auto_rows": null,
3621"grid_column": null,
3622"grid_gap": null,
3623"grid_row": null,
3624"grid_template_areas": null,
3625"grid_template_columns": null,
3626"grid_template_rows": null,
3627"height": null,
3628"justify_content": null,
3629"justify_items": null,
3630"left": null,
3631"margin": null,
3632"max_height": null,
3633"max_width": null,
3634"min_height": null,
3635"min_width": null,
3636"object_fit": null,
3637"object_position": null,
3638"order": null,
3639"overflow": null,
3640"overflow_x": null,
3641"overflow_y": null,
3642"padding": null,
3643"right": null,
3644"top": null,
3645"visibility": null,
3646"width": null
3647}
3648},
3649"f6b507d2bdb74ebaadbbd8a9b51290d6": {
3650"model_module": "@jupyter-widgets/controls",
3651"model_module_version": "1.5.0",
3652"model_name": "HTMLModel",
3653"state": {
3654"_dom_classes": [],
3655"_model_module": "@jupyter-widgets/controls",
3656"_model_module_version": "1.5.0",
3657"_model_name": "HTMLModel",
3658"_view_count": null,
3659"_view_module": "@jupyter-widgets/controls",
3660"_view_module_version": "1.5.0",
3661"_view_name": "HTMLView",
3662"description": "",
3663"description_tooltip": null,
3664"layout": "IPY_MODEL_32b6df9a4f2b42739355c896cea4ca4f",
3665"placeholder": "",
3666"style": "IPY_MODEL_e532f8f1a6cf433fa22410a6876f4707",
3667"value": "Generating train split: 97%"
3668}
3669},
3670"f8cbbb20284f43ea9efcaefc8a790f9a": {
3671"model_module": "@jupyter-widgets/base",
3672"model_module_version": "1.2.0",
3673"model_name": "LayoutModel",
3674"state": {
3675"_model_module": "@jupyter-widgets/base",
3676"_model_module_version": "1.2.0",
3677"_model_name": "LayoutModel",
3678"_view_count": null,
3679"_view_module": "@jupyter-widgets/base",
3680"_view_module_version": "1.2.0",
3681"_view_name": "LayoutView",
3682"align_content": null,
3683"align_items": null,
3684"align_self": null,
3685"border": null,
3686"bottom": null,
3687"display": null,
3688"flex": null,
3689"flex_flow": null,
3690"grid_area": null,
3691"grid_auto_columns": null,
3692"grid_auto_flow": null,
3693"grid_auto_rows": null,
3694"grid_column": null,
3695"grid_gap": null,
3696"grid_row": null,
3697"grid_template_areas": null,
3698"grid_template_columns": null,
3699"grid_template_rows": null,
3700"height": null,
3701"justify_content": null,
3702"justify_items": null,
3703"left": null,
3704"margin": null,
3705"max_height": null,
3706"max_width": null,
3707"min_height": null,
3708"min_width": null,
3709"object_fit": null,
3710"object_position": null,
3711"order": null,
3712"overflow": null,
3713"overflow_x": null,
3714"overflow_y": null,
3715"padding": null,
3716"right": null,
3717"top": null,
3718"visibility": null,
3719"width": null
3720}
3721},
3722"f9e79559b8e34d55b669413423c3c2ae": {
3723"model_module": "@jupyter-widgets/controls",
3724"model_module_version": "1.5.0",
3725"model_name": "FloatProgressModel",
3726"state": {
3727"_dom_classes": [],
3728"_model_module": "@jupyter-widgets/controls",
3729"_model_module_version": "1.5.0",
3730"_model_name": "FloatProgressModel",
3731"_view_count": null,
3732"_view_module": "@jupyter-widgets/controls",
3733"_view_module_version": "1.5.0",
3734"_view_name": "ProgressView",
3735"bar_style": "",
3736"description": "",
3737"description_tooltip": null,
3738"layout": "IPY_MODEL_f5632381613a400f88d61d520f7f27e7",
3739"max": 5452,
3740"min": 0,
3741"orientation": "horizontal",
3742"style": "IPY_MODEL_6b9659cabd534ecb983fc84da293cfdc",
3743"value": 5452
3744}
3745}
3746}
3747}
3748},
3749"nbformat": 4,
3750"nbformat_minor": 0
3751}
3752