Skip to content

Instantly share code, notes, and snippets.

@jeffvestal
Last active January 19, 2023 15:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jeffvestal/6c2023bad4b0476f00b35a6a822fa5b9 to your computer and use it in GitHub Desktop.
Save jeffvestal/6c2023bad4b0476f00b35a6a822fa5b9 to your computer and use it in GitHub Desktop.

Product vector / knn search examples

sentence embedding model - all-mpnet-base-v2 Test dataset - Kaggle


Dataset loaded as is

I used the data uploader in kibana

GET fashion_clothing_products-ori/_search
GET fashion_clothing_products-ori/_mapping
GET fashion_clothing_products-ori/_count

create mapping

PUT fashion_clothing_products-vector
{
      "mappings": {
      "_meta": {
        "created_by": "file-data-visualizer"
      },
      "properties": {
        "Description": {
          "type": "text"
        },
        "Gender": {
          "type": "keyword"
        },
        "NumImages": {
          "type": "long"
        },
        "Price_inr": {
          "type": "long"
        },
        "PrimaryColor": {
          "type": "keyword"
        },
        "ProductBrand": {
          "type": "keyword"
        },
        "ProductID": {
          "type": "long"
        },
        "ProductName": {
          "type": "text"
        },
        "text_embedding": {
          "properties": {
            "is_truncated": {
              "type": "boolean"
            },
            "model_id": {
              "type": "text",
              "fields": {
                "keyword": {
                  "type": "keyword",
                  "ignore_above": 256
                }
              }
            },
            "predicted_value": {
              "type": "dense_vector",
              "dims": 768,
              "index": true,
              "similarity": "dot_product"
            }
          }
        }
      }
    }
}

Reindex to generate vectors

  • You could just do this on ingest also
POST _reindex?wait_for_completion=false
{
  "source": {
    "index": "fashion_clothing_products-ori"
  },
  "dest": {
    "index": "fashion_clothing_products-vector",
    "pipeline": "overstock-all-mpnet-base-v2"
  }
}
GET fashion_clothing_products-vector/_search
GET fashion_clothing_products-vector/_count

search example

generate vector from search query

POST _ml/trained_models/sentence-transformers__all-mpnet-base-v2/_infer
{
  "docs": [
    {
      "text_field": "Rugged red travel suitcase"
    },
    {
       "text_field": "Men's lightweight workout shirt"
    }
  ]
} 

knn _search


Rugged red travel suitcase

Expand for full -> POST fashion_clothing_products-vector/_search json
POST fashion_clothing_products-vector/_search
{
  "_source": {
    "excludes": [
      "text_embedding"
      ]
    }, 
  "knn": {
    "field": "text_embedding.predicted_value",
    "k": 5,
    "num_candidates": 10,
    "query_vector": [
        0.04455750063061714,
        -0.0021901505533605814,
        0.009092449210584164,
        0.0296055655926466,
        0.014209273271262646,
        0.0168576892465353,
        0.007130235433578491,
        0.030451830476522446,
        -0.03164277598261833,
        -0.02236868627369404,
        0.011012016795575619,
        0.026404468342661858,
        -0.0005234265699982643,
        -0.016054630279541016,
        0.020937906578183174,
        0.043481796979904175,
        -0.00211649970151484,
        -0.027513064444065094,
        -0.01588362082839012,
        0.0103794289752841,
        0.0491955503821373,
        -0.01858432963490486,
        -0.0026150024496018887,
        0.040057472884655,
        0.027562951669096947,
        0.0030053514055907726,
        -0.04757470265030861,
        0.05278423801064491,
        -0.058148834854364395,
        -0.0149809829890728,
        0.05472657456994057,
        0.026782816275954247,
        -0.02626628614962101,
        0.0007792466785758734,
        0.000001231366354659258,
        -0.013358689844608307,
        -0.002495075576007366,
        -0.03139485418796539,
        0.01894400641322136,
        -0.09410836547613144,
        0.08294326812028885,
        0.05965275689959526,
        -0.007118603214621544,
        -0.02826770208775997,
        0.03227127343416214,
        -0.071106918156147,
        -0.0433603934943676,
        0.0979727953672409,
        -0.0779222771525383,
        0.028687071055173874,
        0.009905346669256687,
        -0.03674333170056343,
        0.0016611154424026608,
        0.03346358984708786,
        0.06063445284962654,
        -0.019850628450512886,
        0.03045591153204441,
        -0.03310379758477211,
        0.04234981909394264,
        0.017041729763150215,
        -0.04494995251297951,
        -0.04031871631741524,
        0.009092679247260094,
        0.010721676051616669,
        -0.06921617686748505,
        -0.01170361042022705,
        -0.0841069221496582,
        0.016080332919955254,
        -0.03210422024130821,
        0.0302285086363554,
        -0.01681150682270527,
        -0.039722200483083725,
        -0.018490133807063103,
        0.05541589483618736,
        -0.004239013884216547,
        0.004334521945565939,
        0.012589609250426292,
        0.019994737580418587,
        0.012310855090618134,
        0.007561924867331982,
        0.01659809984266758,
        0.019936082884669304,
        -0.027480049058794975,
        0.046768371015787125,
        0.029528791084885597,
        0.01464888360351324,
        -0.00638536037877202,
        0.013977008871734142,
        0.028733599931001663,
        -0.021281417459249496,
        0.0716073289513588,
        0.04549747705459595,
        0.0004659606493078172,
        0.00977130327373743,
        0.0025935692247003317,
        -0.02113146148622036,
        0.010528120212256908,
        -0.0005275350995361805,
        0.028634870424866676,
        -0.04179580509662628,
        0.015257423743605614,
        0.04726774990558624,
        0.024894703179597855,
        0.00789339654147625,
        -0.07550515979528427,
        0.04160626605153084,
        0.030922235921025276,
        -0.009433252736926079,
        -0.048664893954992294,
        0.022959157824516296,
        -0.0360325425863266,
        -0.0065614343620836735,
        -0.04047493636608124,
        -0.0010931288124993443,
        -0.00699825631454587,
        -0.012203242629766464,
        0.04505021870136261,
        -0.023455478250980377,
        0.016783537343144417,
        0.020823141559958458,
        0.02164076268672943,
        -0.03125659376382828,
        -0.039394136518239975,
        -0.012622840702533722,
        0.022364100441336632,
        -0.012013394385576248,
        -0.011178993619978428,
        -0.024594616144895554,
        -0.012111310847103596,
        0.0025753790978342295,
        -0.009927479550242424,
        0.022861313074827194,
        -0.008955300785601139,
        -0.03343256562948227,
        -0.04408923164010048,
        0.03296441584825516,
        -0.022146690636873245,
        -0.032431986182928085,
        -0.02956555224955082,
        -0.004275726620107889,
        0.024310391396284103,
        0.010990461334586143,
        0.05211241543292999,
        -0.007569928653538227,
        -0.0212800782173872,
        -0.010271523147821426,
        0.033913880586624146,
        -0.019239183515310287,
        -0.01586095429956913,
        -0.06817557662725449,
        0.004332040902227163,
        -0.021759498864412308,
        0.05432595685124397,
        0.009290780872106552,
        -0.043268781155347824,
        0.03960992023348808,
        0.04326758533716202,
        0.036311518400907516,
        -0.054971564561128616,
        -0.028842730447649956,
        -0.008024141192436218,
        0.006421975325793028,
        -0.007422258146107197,
        -0.0009521050378680229,
        -0.07612860202789307,
        -0.0063216546550393105,
        -0.06631627678871155,
        -0.023461153730750084,
        -0.00916216243058443,
        -0.027070773765444756,
        -0.03722361847758293,
        -0.0301476139575243,
        0.010710788890719414,
        0.02880050428211689,
        0.018516890704631805,
        -0.0426204651594162,
        0.03968467190861702,
        0.026709791272878647,
        -0.09346941113471985,
        -0.03532715514302254,
        -0.04848106950521469,
        -0.0625041276216507,
        0.06965869665145874,
        -0.00563223147764802,
        -0.03166968375444412,
        -0.025144098326563835,
        -0.03551153838634491,
        -0.0003193424199707806,
        -0.011286788620054722,
        -0.000622245657723397,
        0.010203794576227665,
        0.0059387111105024815,
        0.07521377503871918,
        0.024703551083803177,
        0.0022270961198955774,
        -0.04263671487569809,
        -0.008197715505957603,
        -0.1247028335928917,
        0.0038470146246254444,
        -0.013262137770652771,
        -0.029443103820085526,
        -0.01220334880053997,
        0.08966746181249619,
        -0.004999502561986446,
        -0.023497233167290688,
        -0.004671940580010414,
        -0.010621700435876846,
        0.014125289395451546,
        0.08393171429634094,
        -0.002235070802271366,
        0.07293146848678589,
        0.04009638726711273,
        -0.003992861602455378,
        -0.06185324862599373,
        -0.05961129069328308,
        0.011165760457515717,
        0.010787203907966614,
        -0.020915953442454338,
        -0.005905483849346638,
        0.019318968057632446,
        0.024565696716308594,
        -0.0036066065076738596,
        0.021129071712493896,
        -0.026613911613821983,
        0.04603219032287598,
        0.01303822360932827,
        0.02931218408048153,
        -0.014870969578623772,
        0.01767713762819767,
        -0.0024867451284080744,
        0.0062749795615673065,
        -0.0021926071494817734,
        -0.0443563349545002,
        -0.020984522998332977,
        0.03110140562057495,
        0.026567203924059868,
        0.1021348237991333,
        -0.014147154055535793,
        -0.05231975018978119,
        0.0407952144742012,
        -0.039179302752017975,
        -0.015660958364605904,
        -0.006088758818805218,
        0.022863343358039856,
        -0.04594032093882561,
        0.0739787220954895,
        0.028776004910469055,
        -0.013986188918352127,
        0.0022076808381825686,
        -0.017230672761797905,
        0.04363144189119339,
        -0.015963446348905563,
        0.016077181324362755,
        -0.021906375885009766,
        0.029585786163806915,
        -0.05051020532846451,
        -0.042173728346824646,
        0.025723520666360855,
        -0.011911971494555473,
        -0.015672480687499046,
        0.009909059852361679,
        0.013203145936131477,
        0.005964836571365595,
        0.027791835367679596,
        0.007581279147416353,
        -0.0033736731857061386,
        0.025674991309642792,
        0.09987873584032059,
        -0.018031029030680656,
        -0.012787661515176296,
        -0.026005221530795097,
        0.012957732193171978,
        0.014347857795655727,
        0.005563822574913502,
        0.010858345776796341,
        -0.056361064314842224,
        0.017246657982468605,
        -0.005178908351808786,
        -0.02717522159218788,
        -0.0047548129223287106,
        -0.025712205097079277,
        0.02497546374797821,
        -0.04359547048807144,
        -0.002688709646463394,
        0.02142118290066719,
        0.028278620913624763,
        -0.04336971044540405,
        0.06605930626392365,
        -0.061293721199035645,
        -0.021709825843572617,
        -0.01582685299217701,
        0.009477225132286549,
        -0.010368023999035358,
        -0.05768774822354317,
        -0.05516796559095383,
        -0.05595443770289421,
        0.05161391571164131,
        -0.005081977229565382,
        0.04881223291158676,
        -0.09325595200061798,
        0.0066609736531972885,
        -0.04336871579289436,
        -0.025093551725149155,
        -0.004703802522271872,
        0.008113767020404339,
        0.0006707843276672065,
        0.0073290374130010605,
        -0.008609608747065067,
        -0.03437764570116997,
        -0.04684818908572197,
        -0.023757213726639748,
        -0.02213439531624317,
        0.019415786489844322,
        -0.06730847805738449,
        0.0478946827352047,
        -0.054552607238292694,
        0.02740565314888954,
        0.007350120693445206,
        0.0042699771001935005,
        0.03913732245564461,
        -0.04597677290439606,
        0.029313726350665092,
        0.017158426344394684,
        -0.016462037339806557,
        -0.010813937522470951,
        -0.029181359335780144,
        -0.03318340331315994,
        -0.017128828912973404,
        -0.017551470547914505,
        0.029538851231336594,
        0.015511348843574524,
        -0.09341712296009064,
        0.07599513977766037,
        0.04673662409186363,
        -0.037826504558324814,
        -0.10542672872543335,
        -0.036197248846292496,
        -0.052141472697257996,
        0.012692016549408436,
        -0.008516398258507252,
        -0.02010931260883808,
        -0.06610371917486191,
        0.02333308383822441,
        -0.0025044172070920467,
        -0.024204904213547707,
        -0.03008723072707653,
        -0.023900019004940987,
        0.04923361539840698,
        0.0027859725523740053,
        0.020781975239515305,
        0.02915560081601143,
        0.027084846049547195,
        0.04198293760418892,
        0.005617238115519285,
        0.015711627900600433,
        -0.0067846523597836494,
        -0.0016574100591242313,
        0.05216106399893761,
        0.04294133931398392,
        -0.0009119502501562238,
        0.004959430079907179,
        -0.011414744891226292,
        -0.011152639053761959,
        -0.03386370465159416,
        -0.037507712841033936,
        0.03599398210644722,
        0.023694148287177086,
        0.07942579686641693,
        0.013575694523751736,
        0.009205947630107403,
        0.012044521048665047,
        0.031841617077589035,
        0.01878982223570347,
        0.006634016986936331,
        -0.014969397336244583,
        -0.026368502527475357,
        -0.013095150701701641,
        -0.019652586430311203,
        0.04873434454202652,
        0.010767682455480099,
        0.06038330867886543,
        0.014243200421333313,
        -0.020540446043014526,
        0.02437479794025421,
        -0.04714815318584442,
        -0.0675659030675888,
        -0.04157985374331474,
        -0.023009393364191055,
        -0.012893527746200562,
        -0.02455752156674862,
        0.03472725301980972,
        0.07604088634252548,
        0.03726090490818024,
        0.05516933649778366,
        -0.010427654720842838,
        0.040272027254104614,
        -0.007595094386488199,
        0.024801047518849373,
        -0.014513625763356686,
        0.010430854745209217,
        -0.011553822085261345,
        0.004813984967768192,
        0.03724932670593262,
        0.020901955664157867,
        -0.0011528526665642858,
        -0.010091986507177353,
        -0.07383183389902115,
        0.022566966712474823,
        -0.020754097029566765,
        -0.03631797060370445,
        0.001472330535762012,
        0.036924008280038834,
        0.005391320213675499,
        0.04243423044681549,
        -0.11831692606210709,
        -0.002176434500142932,
        -0.03458654135465622,
        -0.049030400812625885,
        -0.056045860052108765,
        -0.028249429538846016,
        0.00047018524492159486,
        -0.007178972475230694,
        -0.0032038791105151176,
        -0.0007654322544112802,
        0.031618501991033554,
        -0.01838536746799946,
        -0.09939857572317123,
        -0.03084423579275608,
        0.005209504161030054,
        0.0001532860769657418,
        0.029931027442216873,
        -0.07497277855873108,
        -0.007238625083118677,
        0.050276368856430054,
        -0.04056129232048988,
        0.006708413362503052,
        0.03363099694252014,
        -0.028225496411323547,
        0.04583165422081947,
        -0.01014766376465559,
        0.020225409418344498,
        0.017979556694626808,
        -0.03920074924826622,
        -0.01650092750787735,
        -0.0022571429144591093,
        -0.16800786554813385,
        0.017744753509759903,
        -0.019516363739967346,
        -0.023193607106804848,
        -0.021128162741661072,
        0.011926740407943726,
        -0.023894140496850014,
        0.06242531165480614,
        0.009923537261784077,
        -0.0554518885910511,
        -0.0007250737398862839,
        0.1150919646024704,
        0.07285358011722565,
        -0.00914587639272213,
        0.04733375087380409,
        0.006650998722761869,
        0.012030469253659248,
        -0.058223944157361984,
        0.029148047789931297,
        -0.020076550543308258,
        -0.014804004691541195,
        -0.04568042233586311,
        -0.010926957242190838,
        -0.005281292833387852,
        0.005280087701976299,
        -0.027413783594965935,
        -0.03686216101050377,
        -0.020834224298596382,
        -0.010984291322529316,
        0.023375287652015686,
        -0.0387638695538044,
        -0.00009265671542380005,
        -0.020374704152345657,
        0.0027464109007269144,
        0.0159956905990839,
        -0.0553322359919548,
        0.03233732283115387,
        0.04688088968396187,
        0.030838562175631523,
        -0.005281498655676842,
        0.055466845631599426,
        -0.007633334957063198,
        -0.015473459847271442,
        0.005345009732991457,
        -0.026694966480135918,
        0.012034432962536812,
        0.002525215269997716,
        0.0012910707155242562,
        0.05750637128949165,
        0.031173354014754295,
        0.06308402121067047,
        0.020289061591029167,
        -0.0053083328530192375,
        0.03818218782544136,
        -0.0268891341984272,
        0.030969195067882538,
        0.023162486031651497,
        -0.03130228817462921,
        0.045378778129816055,
        0.014845510013401508,
        -0.046134259551763535,
        -0.0010116846533492208,
        0.03923534229397774,
        -0.002646464854478836,
        -0.015858246013522148,
        0.0341683104634285,
        0.0213252454996109,
        0.018386689946055412,
        -0.036428824067115784,
        0.018928105011582375,
        0.07323092222213745,
        -0.062130093574523926,
        -0.04282310605049133,
        -0.05132631957530975,
        0.0036549558863043785,
        0.016962168738245964,
        -0.03954242169857025,
        0.006727301515638828,
        -0.01286470890045166,
        -0.03496970608830452,
        0.03092276304960251,
        0.04217815771698952,
        -0.00245282961986959,
        0.03783801198005676,
        -0.029023610055446625,
        0.04549318179488182,
        -0.052024710923433304,
        0.0010879043256863952,
        0.0153959384188056,
        -0.04505284130573273,
        0.09737935662269592,
        -0.015354549512267113,
        0.04208526760339737,
        -0.01116067823022604,
        -0.0030842148698866367,
        -0.03817710280418396,
        -0.009443560615181923,
        0.006491553038358688,
        0.012357364408671856,
        -0.02987283281981945,
        0.04116944968700409,
        -0.03661777824163437,
        0.013704636134207249,
        -0.014037646353244781,
        0.001220973557792604,
        -0.015366381034255028,
        -0.05666901171207428,
        -0.08796723932027817,
        -0.016073234379291534,
        -0.009126588702201843,
        -4.646617864179159e-33,
        0.05270428955554962,
        -0.011843826621770859,
        -0.03635844215750694,
        0.02842988818883896,
        0.06040965020656586,
        0.033967141062021255,
        -0.01691477745771408,
        0.019186297431588173,
        0.002094890456646681,
        -0.011051694862544537,
        0.015268256887793541,
        -0.07179892063140869,
        0.019517354667186737,
        0.028071513399481773,
        0.020870214328169823,
        0.012090983800590038,
        0.017085719853639603,
        0.06437542289495468,
        0.03441030904650688,
        -0.05136541277170181,
        0.008098539896309376,
        0.018007224425673485,
        0.0429261215031147,
        0.03503093123435974,
        -0.037135425955057144,
        -0.023113569244742393,
        0.03367997705936432,
        0.006249573547393084,
        0.056205034255981445,
        0.045051269233226776,
        -0.015235062688589096,
        0.016437936574220657,
        0.010247517377138138,
        0.039342209696769714,
        0.036953773349523544,
        0.014044376090168953,
        0.009378249756991863,
        -0.03925274685025215,
        0.04209583252668381,
        0.011246435344219208,
        -0.022633543238043785,
        0.011425429955124855,
        0.029204895719885826,
        -0.031777746975421906,
        0.011463232338428497,
        0.018863579258322716,
        -0.024690454825758934,
        0.01619330234825611,
        0.005213030613958836,
        0.037653617560863495,
        0.023814864456653595,
        0.023847637698054314,
        -0.003020269563421607,
        -0.002974907634779811,
        0.04173683002591133,
        -0.061352603137493134,
        0.028546355664730072,
        0.04709345102310181,
        0.029647458344697952,
        -0.03477753326296806,
        -0.08687940239906311,
        -0.007014830131083727,
        -0.02770068123936653,
        0.030740240588784218,
        -0.0385216549038887,
        0.01773403212428093,
        0.0316314697265625,
        0.01735575869679451,
        -0.04727969318628311,
        -0.022414445877075195,
        -0.004910692106932402,
        -0.046476997435092926,
        0.017516963183879852,
        0.07354333251714706,
        -0.04945793002843857,
        0.018527435138821602,
        0.01057475060224533,
        -0.04051647707819939,
        0.11401987075805664,
        0.0611882247030735,
        0.018183043226599693,
        0.048523884266614914,
        -0.004711741115897894,
        0.04043729230761528,
        -0.021109748631715775,
        0.04129024222493172,
        -0.008208244107663631,
        0.005339090712368488,
        0.024966971948742867,
        0.03264700248837471,
        -0.011139927431941032,
        0.03475990518927574,
        0.008400000631809235,
        0.05765380710363388,
        -0.030428718775510788,
        0.027948740869760513,
        0.011942273937165737,
        -0.005342045333236456,
        -0.022235466167330742,
        -0.02362227812409401,
        0.049535054713487625,
        0.053310271352529526,
        -0.020012542605400085,
        -0.08498235791921616,
        0.008196845650672913,
        -0.03499150648713112,
        0.008056329563260078,
        0.018663061782717705,
        -0.0371055081486702,
        0.009746365249156952,
        0.01502423919737339,
        -0.009601147845387459,
        -0.020209671929478645,
        0.003048879560083151,
        -0.023223109543323517,
        -0.025595882907509804,
        0.019420268014073372,
        0.012449230067431927,
        -0.0005818400532007217,
        -0.04244381934404373,
        0.011258810758590698,
        -0.02745557203888893,
        0.019838502630591393,
        -0.0311879925429821,
        -0.051971495151519775,
        -0.022393222898244858,
        0.004664967767894268,
        0.004922336898744106,
        0.04058514162898064,
        0.012584165669977665,
        0.033672891557216644,
        0.024368982762098312,
        1.9525893435456965e-7,
        0.045695945620536804,
        -0.006561854854226112,
        -0.008791900239884853,
        -0.07309366017580032,
        -0.011461453512310982,
        -0.01700102910399437,
        -0.034624133259058,
        0.023307260125875473,
        0.03815685585141182,
        -0.03325297683477402,
        -0.02688334323465824,
        0.007727762684226036,
        -0.021474523469805717,
        0.018713077530264854,
        0.025508278980851173,
        0.028348812833428383,
        0.04301903024315834,
        -0.041773173958063126,
        -0.049639277160167694,
        0.030772635713219643,
        0.0468033030629158,
        0.026827024295926094,
        0.03520475700497627,
        -0.023558394983410835,
        -0.026165932416915894,
        -0.061000894755125046,
        0.017264459282159805,
        -0.07663245499134064,
        0.054247647523880005,
        -0.06191631034016609,
        -0.0011316585587337613,
        0.0020938522648066282,
        -0.018887104466557503,
        -0.007015176583081484,
        0.006718161515891552,
        -0.025130221620202065,
        0.019251680001616478,
        0.08374378085136414,
        0.06381266564130783,
        -0.046447619795799255,
        -0.006685527507215738,
        0.016825396567583084,
        -0.04304647445678711,
        -0.03546180576086044,
        0.03796421363949776,
        0.07749371230602264,
        0.027755947783589363,
        0.030074436217546463,
        -0.027150718495249748,
        -0.0033814890775829554,
        0.06015101820230484,
        -0.012707452289760113,
        -0.002177078975364566,
        0.017911609262228012,
        -0.028547603636980057,
        0.005972757004201412,
        0.026172300800681114,
        -0.03827616199851036,
        -0.010421762242913246,
        -0.04608101025223732,
        -0.07006298750638962,
        0.08825455605983734,
        0.014790846966207027,
        -0.007615965325385332,
        0.014775218442082405,
        -0.09015641361474991,
        0.052920568734407425,
        8.791249613170669e-35,
        0.0000408500200137496,
        -0.008390874601900578,
        -0.0013467208482325077,
        0.005779595579952002,
        0.00616774708032608,
        0.01553182303905487,
        0.04846412315964699,
        0.007659324910491705,
        0.019292237237095833,
        -0.02227829582989216,
        -0.01777804084122181
      ]
  }
} 

Men's lightweight workout shirt

Expand for full -> POST fashion_clothing_products-vector/_search json
POST fashion_clothing_products-vector/_search
{
  "_source": {
    "excludes": [
      "text_embedding"
      ]
    }, 
  "knn": {
    "field": "text_embedding.predicted_value",
    "k": 5,
    "num_candidates": 10,
    "query_vector": [
        -0.024684038013219833,
        -0.030416471883654594,
        0.01845187321305275,
        -0.006987069733440876,
        -0.04328552260994911,
        -0.029441284015774727,
        0.04453010484576225,
        0.07273009419441223,
        -0.04487353935837746,
        0.005814062897115946,
        0.047535959631204605,
        -0.013783073984086514,
        0.0033103222958743572,
        0.04447850584983826,
        0.059871427714824677,
        0.02428683079779148,
        -0.0718807578086853,
        0.023662863299250603,
        0.056765444576740265,
        -0.029420269653201103,
        0.012038824148476124,
        -0.006247951649129391,
        0.02746064029633999,
        0.016106732189655304,
        0.04840905964374542,
        -0.034325193613767624,
        0.021205658093094826,
        0.0010782578028738499,
        -0.03898018226027489,
        -0.024093996733427048,
        0.029430177062749863,
        0.005048342514783144,
        -0.031428780406713486,
        0.02463582344353199,
        0.0000011595760724958382,
        -0.006527545861899853,
        0.06553450971841812,
        -0.012456142343580723,
        0.007398295681923628,
        -0.025883695110678673,
        -0.005775296129286289,
        0.0663016140460968,
        -0.010800047777593136,
        -0.03847675025463104,
        0.002862593624740839,
        -0.01277155615389347,
        0.038832034915685654,
        0.04312937334179878,
        -0.06888794153928757,
        -0.015164465643465519,
        0.025537768378853798,
        0.04573698714375496,
        -0.03481123223900795,
        -0.0073060342110693455,
        0.0042497641406953335,
        0.03439706563949585,
        -0.0026982722338289022,
        -0.02985888347029686,
        0.03134298324584961,
        0.07397576421499252,
        0.026541298255324364,
        -0.030309537425637245,
        -0.012384319677948952,
        0.007810445502400398,
        0.008365917019546032,
        -0.01829647831618786,
        -0.028173008933663368,
        -0.012092254124581814,
        -0.005607864819467068,
        0.08808273077011108,
        -0.00782214105129242,
        -0.01300289947539568,
        0.017920779064297676,
        0.030928339809179306,
        0.05913170799612999,
        -0.00610488373786211,
        -0.058997903019189835,
        -0.05876713618636131,
        0.002152630826458335,
        0.007178702391684055,
        -0.06964416801929474,
        0.0037794262170791626,
        -0.030918728560209274,
        0.053919531404972076,
        -0.014091813936829567,
        -0.0456002838909626,
        0.009868592023849487,
        -0.036221593618392944,
        0.023415179923176765,
        -0.008049114607274532,
        0.02094828523695469,
        -0.013579746708273888,
        -0.025832757353782654,
        -0.009925903752446175,
        -0.010147074237465858,
        0.003378440160304308,
        0.003805591491982341,
        -0.07464572042226791,
        0.061637066304683685,
        -0.04007689282298088,
        -0.06206488236784935,
        -0.008427781984210014,
        0.03459525108337402,
        0.0035172435455024242,
        -0.01284191943705082,
        0.011209133081138134,
        0.034004516899585724,
        -0.024670982733368874,
        0.004837269429117441,
        0.02880941517651081,
        -0.0054696123115718365,
        0.043053578585386276,
        -0.05446852371096611,
        0.07492285966873169,
        0.04398996755480766,
        -0.059488147497177124,
        -0.07864271849393845,
        -0.000902174215298146,
        -0.012147350236773491,
        0.023539306595921516,
        0.03182237595319748,
        0.04956452175974846,
        0.008518910966813564,
        0.011178312823176384,
        0.041111428290605545,
        0.0008945466834120452,
        0.0033720217179507017,
        0.02996133640408516,
        0.0013399766758084297,
        -0.03248251602053642,
        -0.019639622420072556,
        0.03175156190991402,
        0.0023494907654821873,
        -0.036871105432510376,
        0.0019868032541126013,
        -0.018801096826791763,
        -0.011367845349013805,
        -0.03460482135415077,
        0.004972212016582489,
        -0.0329703651368618,
        -0.01902671717107296,
        0.07414199411869049,
        -0.017017649486660957,
        -0.0019136433256790042,
        0.011946206912398338,
        -0.023251140490174294,
        0.029215848073363304,
        0.03931891918182373,
        0.015137062408030033,
        0.03628624230623245,
        -0.03164326027035713,
        -0.003913115244358778,
        0.07517454773187637,
        0.024384157732129097,
        -0.01140279695391655,
        0.0021358320955187082,
        -0.016320550814270973,
        0.023866618052124977,
        -0.026010608300566673,
        0.016067203134298325,
        -0.004322351422160864,
        -0.013340212404727936,
        -0.07773567736148834,
        -0.007983139716088772,
        0.0035346760414540768,
        -0.021767878904938698,
        -0.07174564152956009,
        -0.013469868339598179,
        -0.0088101327419281,
        0.04036218672990799,
        -0.030075104907155037,
        -0.04663412272930145,
        -0.028076212853193283,
        0.022981690242886543,
        0.09155460447072983,
        -0.022390367463231087,
        0.01976635493338108,
        -0.016519322991371155,
        -0.036825135350227356,
        0.04010900855064392,
        -0.03653346002101898,
        -0.060660697519779205,
        -0.03241074085235596,
        0.0034179268404841423,
        -0.029781393706798553,
        0.007575022988021374,
        -0.017940973863005638,
        -0.021874845027923584,
        -0.03015744499862194,
        0.0038073945324867964,
        0.031687505543231964,
        -0.05696035549044609,
        -0.013063625432550907,
        0.004255053121596575,
        0.05318336561322212,
        0.00575204798951745,
        -0.04337933659553528,
        0.034736961126327515,
        -0.003678811015561223,
        -0.019635826349258423,
        -0.05775435268878937,
        0.012431174516677856,
        -0.04751087725162506,
        -0.04766325652599335,
        -0.045040030032396317,
        0.019282376393675804,
        0.019550561904907227,
        0.04733461141586304,
        0.029003890231251717,
        0.010609901510179043,
        0.005340122152119875,
        0.035594113171100616,
        -0.02486483007669449,
        -0.06989891827106476,
        -0.03794131055474281,
        -0.009768516756594181,
        -0.005923509132117033,
        0.07969669997692108,
        -0.07854179292917252,
        -0.059858281165361404,
        0.02062216028571129,
        0.0046034022234380245,
        -0.0178841520100832,
        -0.0051505775190889835,
        0.006785950623452663,
        0.0249788761138916,
        -0.001976638101041317,
        0.017777932807803154,
        0.03768881782889366,
        0.022176014259457588,
        -0.03963533788919449,
        -0.03580441325902939,
        0.023153183981776237,
        0.015715966001152992,
        -0.07320591062307358,
        0.04224252700805664,
        0.0444304458796978,
        0.031831108033657074,
        -0.008550869300961494,
        0.013234640471637249,
        0.029036905616521835,
        -0.020453957840800285,
        0.02572454884648323,
        -0.07721374928951263,
        0.003756773192435503,
        0.010311947204172611,
        -0.012822918593883514,
        -0.013775559142231941,
        -0.08596426993608475,
        0.04280365630984306,
        -0.05404546111822128,
        0.03596760332584381,
        0.02246340736746788,
        0.012637701816856861,
        -0.0009519587620161474,
        0.023616015911102295,
        -0.05502777546644211,
        0.03729698807001114,
        -0.02914646454155445,
        0.002757920417934656,
        0.01709897443652153,
        -0.07139507681131363,
        0.02275678515434265,
        0.02291794680058956,
        0.04998229816555977,
        0.012877234257757664,
        -0.012057797983288765,
        0.08093073219060898,
        0.06226303055882454,
        -0.0010919277556240559,
        -0.0060758828185498714,
        -0.001509364927187562,
        0.008615502156317234,
        -0.005468299612402916,
        -0.04190583899617195,
        -0.06454849243164062,
        0.03984494134783745,
        -0.05137140303850174,
        -0.016743963584303856,
        -0.01494833268225193,
        0.027238845825195312,
        0.07053545862436295,
        0.042092204093933105,
        0.04927409812808037,
        0.04391760379076004,
        0.0035200081765651703,
        -0.02864665724337101,
        0.005414813756942749,
        -0.015190131030976772,
        0.07132305949926376,
        -0.021986082196235657,
        -0.02516261488199234,
        0.021548444405198097,
        -0.06271526217460632,
        0.04554653912782669,
        0.03236236423254013,
        0.01955067552626133,
        -0.03231072053313255,
        0.000703647208865732,
        -0.09394437074661255,
        -0.0012615605955943465,
        0.026011331006884575,
        0.0046536861918866634,
        -0.025871777907013893,
        -0.023704199120402336,
        -0.02443978190422058,
        0.05400867760181427,
        -0.05123766511678696,
        0.026810551062226295,
        0.018042398616671562,
        -0.009318661876022816,
        -0.028328653424978256,
        -0.01971224509179592,
        -0.03229886665940285,
        -0.01967693492770195,
        0.009157614782452583,
        -0.027890339493751526,
        0.07922647148370743,
        0.004945275839418173,
        -0.008684958331286907,
        -0.01259898766875267,
        -0.00038986268918961287,
        -0.01653292030096054,
        -0.011495429091155529,
        -0.013702833093702793,
        0.010302742943167686,
        -0.03209761530160904,
        -0.012077532708644867,
        -0.10509385168552399,
        0.031942032277584076,
        -0.018543943762779236,
        0.0244484543800354,
        -0.01892872154712677,
        0.01588863506913185,
        0.06491107493638992,
        -0.06314276903867722,
        -0.024707498028874397,
        -0.0478719100356102,
        0.0003668363206088543,
        -0.01138676144182682,
        -0.015919016674160957,
        -0.06095815822482109,
        0.06004462391138077,
        0.0019153364701196551,
        0.01714000292122364,
        0.010510144755244255,
        0.010535449720919132,
        0.04445832595229149,
        -0.042006928473711014,
        0.009878828190267086,
        -0.06696018576622009,
        -0.06698017567396164,
        -0.0027225022204220295,
        0.0032657005358487368,
        -0.047059591859579086,
        0.02923079952597618,
        -0.04296921566128731,
        0.01891910657286644,
        -0.00032903230749070644,
        0.006468924228101969,
        -0.029624873772263527,
        0.026987727731466293,
        -0.02398580312728882,
        -0.0007925406680442393,
        -0.030602434650063515,
        -0.030086835846304893,
        0.02105545997619629,
        0.01445509772747755,
        -0.011832908727228642,
        -0.049426037818193436,
        0.06607994437217712,
        0.054984234273433685,
        -0.010883964598178864,
        0.024904362857341766,
        -0.00581448245793581,
        0.000022466667360276915,
        -0.0028981014620512724,
        0.0125169288367033,
        0.0717114508152008,
        0.06219994276762009,
        -0.07782748341560364,
        0.006527301389724016,
        0.01818908005952835,
        -0.022725578397512436,
        0.05117351934313774,
        -0.00915207527577877,
        -0.010421942919492722,
        -0.006501460447907448,
        0.05756587162613869,
        -0.024129793047904968,
        0.02442094497382641,
        0.05877113714814186,
        0.04587346687912941,
        -0.023709891363978386,
        0.040379445999860764,
        -0.016817990690469742,
        -0.014368852600455284,
        -0.03581111505627632,
        0.01659819856286049,
        0.0517590306699276,
        0.04168817400932312,
        0.07172621041536331,
        -0.01765093207359314,
        0.00924946740269661,
        0.017325133085250854,
        -0.014868013560771942,
        0.03578456491231918,
        0.05967743694782257,
        -0.06748190522193909,
        -0.005997576750814915,
        0.02307705022394657,
        -0.03971899300813675,
        -0.05690290033817291,
        -0.0025806252378970385,
        -0.02365759387612343,
        0.007501768413931131,
        0.002347572473809123,
        -0.028894783928990364,
        0.052711568772792816,
        0.017900917679071426,
        0.05489341542124748,
        0.031411245465278625,
        0.002578124636784196,
        0.025193024426698685,
        0.041189104318618774,
        -0.015252286568284035,
        -0.034130699932575226,
        0.014509755186736584,
        -0.02198774367570877,
        -0.022189488634467125,
        -0.029532942920923233,
        0.00039816799107939005,
        -0.058982256799936295,
        0.04864712059497833,
        -0.03297513350844383,
        0.013835988938808441,
        -0.00977316778153181,
        -0.051168009638786316,
        0.06725290417671204,
        0.09189055114984512,
        -0.04671916738152504,
        -0.030990906059741974,
        -0.04584263637661934,
        -0.02218940667808056,
        0.040715474635362625,
        -0.03841288015246391,
        0.02347278781235218,
        -0.01039545051753521,
        -0.06496652215719223,
        0.03164165839552879,
        -0.007429980207234621,
        -0.025102093815803528,
        -0.01711796596646309,
        0.010189637541770935,
        0.006841899827122688,
        0.006068085320293903,
        0.03653635457158089,
        -0.030188100412487984,
        -0.0015623238869011402,
        -0.05101097375154495,
        0.0265610683709383,
        0.007619263604283333,
        -0.12510687112808228,
        0.0161837637424469,
        0.021490931510925293,
        -0.013283873908221722,
        0.03514876216650009,
        -0.0025539062917232513,
        0.0512269027531147,
        -0.011402458883821964,
        -0.02591000869870186,
        -0.009407121688127518,
        0.07515974342823029,
        -0.020247498527169228,
        0.015322958119213581,
        0.011275269091129303,
        0.08569527417421341,
        -0.023884505033493042,
        0.004017097409814596,
        0.06262614578008652,
        0.0036945983301848173,
        -0.004051707219332457,
        -0.030293133109807968,
        0.004951625596731901,
        -0.015548105351626873,
        0.024283884093165398,
        0.028831545263528824,
        0.005423033144325018,
        -0.05333368852734566,
        0.026598429307341576,
        0.022668052464723587,
        0.015442037023603916,
        -0.00602070614695549,
        -0.010853166691958904,
        0.02567438594996929,
        0.11337915807962418,
        0.012596705928444862,
        -0.016160067170858383,
        -0.039360303431749344,
        0.059954933822155,
        0.049457136541604996,
        -0.046047065407037735,
        0.0731143206357956,
        -0.011425682343542576,
        -0.024947751313447952,
        -0.0722866877913475,
        -0.020983874797821045,
        0.011980450712144375,
        0.008070836775004864,
        -0.011299476027488708,
        0.010711860843002796,
        -0.013158987276256084,
        0.047654036432504654,
        0.0573570616543293,
        -0.039152275770902634,
        -0.020356275141239166,
        0.0002013932098634541,
        0.033274151384830475,
        0.027394579723477364,
        0.03208885341882706,
        0.00713602127507329,
        -0.007584382779896259,
        0.06003193184733391,
        -0.007845258340239525,
        -0.056658785790205,
        0.033678725361824036,
        -0.01732352189719677,
        0.003709250595420599,
        0.013844666071236134,
        0.04632124304771423,
        0.011887720786035061,
        -0.03427448868751526,
        0.04948355257511139,
        -0.05522073805332184,
        -0.018312225118279457,
        0.06369643658399582,
        -0.00652435515075922,
        0.06720064580440521,
        0.03544684126973152,
        -0.02105025015771389,
        -0.06315156817436218,
        0.01011685747653246,
        0.025582876056432724,
        -0.018775546923279762,
        0.07014176994562149,
        -0.007219142280519009,
        -0.01301917526870966,
        0.02240988053381443,
        -0.03826479986310005,
        -0.008219771087169647,
        0.02533245086669922,
        0.049042511731386185,
        -0.02699539251625538,
        -0.031949713826179504,
        0.03572649136185646,
        -4.674965645133208e-33,
        -0.017154086381196976,
        -0.01944817788898945,
        0.0028604581020772457,
        -0.0735449567437172,
        0.0033938672859221697,
        0.08318032324314117,
        0.017942897975444794,
        -0.016251863911747932,
        -0.01594085991382599,
        -0.03350391983985901,
        0.021269721910357475,
        0.0076779453083872795,
        -0.010795348323881626,
        -0.007190498989075422,
        0.006167314946651459,
        0.025282397866249084,
        -0.030169498175382614,
        -0.04227316379547119,
        0.06872261315584183,
        0.0043350448831915855,
        -0.029201528057456017,
        -0.018759554252028465,
        -0.003763065440580249,
        0.0024393275380134583,
        0.016290243715047836,
        0.06340338289737701,
        0.01650872640311718,
        -0.07990138977766037,
        -0.02947176620364189,
        -0.003349240403622389,
        -0.013928588479757309,
        0.035260848701000214,
        -0.05646836385130882,
        0.017505981028079987,
        0.038281336426734924,
        -0.005520976614207029,
        0.03389666974544525,
        0.021045396104454994,
        -0.044682495296001434,
        -0.007209770381450653,
        0.03367065265774727,
        -0.01591917872428894,
        -0.01161554828286171,
        -0.028676837682724,
        -0.030953550711274147,
        -0.007353033404797316,
        0.008765141479671001,
        0.031067416071891785,
        -0.02638201229274273,
        -0.03092818520963192,
        0.029125843197107315,
        0.01710631139576435,
        0.0012787241721525788,
        0.023709304630756378,
        0.00994053017348051,
        -0.011581684462726116,
        0.008846824988722801,
        0.006430278066545725,
        -0.06341899186372757,
        0.047349054366350174,
        0.06306061148643494,
        -0.025181857869029045,
        -0.0398261696100235,
        0.021346459165215492,
        -0.026084451004862785,
        -0.0864575207233429,
        0.006839047651737928,
        0.03817766159772873,
        0.018166525289416313,
        -0.01706080511212349,
        -0.03917596489191055,
        0.013847403228282928,
        0.05594316124916077,
        -0.07357804477214813,
        0.027839479967951775,
        -0.007724569644778967,
        -0.03106672130525112,
        -0.01644471287727356,
        0.041082657873630524,
        0.05144164711236954,
        -0.03327515721321106,
        0.04925396665930748,
        0.004017855506390333,
        0.029327258467674255,
        -0.024841491132974625,
        -0.08171442151069641,
        -0.006905860733240843,
        -0.06136279180645943,
        0.007669941056519747,
        -0.024513492360711098,
        0.014557513408362865,
        -0.029730021953582764,
        0.012126696296036243,
        0.02682643011212349,
        -0.035470861941576004,
        -0.03024178370833397,
        0.03145796060562134,
        0.05331582948565483,
        -0.0002600045991130173,
        -0.036068934947252274,
        -0.02397814579308033,
        0.015239104628562927,
        -0.0070806751027703285,
        0.021174656227231026,
        -0.02524777501821518,
        -0.04586658999323845,
        0.05901821330189705,
        0.009907616302371025,
        0.0043240441009402275,
        -0.010487777180969715,
        -0.014299910515546799,
        -0.038447219878435135,
        -0.003534607356414199,
        -0.04891224578022957,
        0.08188432455062866,
        -0.00007127642311388627,
        0.02541126310825348,
        0.02213326096534729,
        0.009831014089286327,
        -0.08398708701133728,
        -0.02682577073574066,
        -0.014837470836937428,
        0.01409248635172844,
        -0.030792877078056335,
        -0.004506648052483797,
        -0.0330558717250824,
        0.0628376454114914,
        0.038399845361709595,
        0.015070772729814053,
        -0.04368402436375618,
        -0.03971043974161148,
        -0.02238406240940094,
        1.9399836048705765e-7,
        -0.048017874360084534,
        -0.02971547842025757,
        -0.009370442479848862,
        -0.01650528609752655,
        -0.043971460312604904,
        0.019211063161492348,
        0.041056834161281586,
        0.0626310408115387,
        -0.004771633073687553,
        -0.035992030054330826,
        0.05762271583080292,
        -0.03777618333697319,
        -0.040162745863199234,
        -0.022448286414146423,
        0.038060691207647324,
        -0.048171110451221466,
        0.014806593768298626,
        -0.04848143830895424,
        0.025000717490911484,
        0.024370715022087097,
        0.029597122222185135,
        0.01618625968694687,
        -0.004758037626743317,
        0.015366649255156517,
        0.008631527423858643,
        -0.034266483038663864,
        -0.00454793032258749,
        0.009200525470077991,
        0.03234807401895523,
        -0.015276262536644936,
        0.03959601745009422,
        -0.02428697794675827,
        -0.007907786406576633,
        -0.020852170884609222,
        0.0032651862129569054,
        -0.002581267151981592,
        0.059417955577373505,
        0.05968928709626198,
        -0.03831247240304947,
        0.040202848613262177,
        -0.031097711995244026,
        0.02678930200636387,
        -0.029653409495949745,
        0.03464370220899582,
        0.005061672069132328,
        -0.004209175240248442,
        -0.0034304100554436445,
        0.07771722972393036,
        -0.05911847949028015,
        0.021106038242578506,
        0.004979349207133055,
        -0.04496608301997185,
        0.08829725533723831,
        0.010534651577472687,
        -0.005005522631108761,
        0.024972304701805115,
        -0.034552428871393204,
        0.02548951655626297,
        0.035170093178749084,
        -0.02283990941941738,
        -0.008832953870296478,
        0.03729937598109245,
        -0.0640515610575676,
        0.09298046678304672,
        -0.00541161373257637,
        -0.05312633886933327,
        -0.016739876940846443,
        1.052833352023905e-34,
        -0.04967467486858368,
        -0.09106579422950745,
        0.00015904287283774465,
        0.020679669454693794,
        -0.009059029631316662,
        0.029375070706009865,
        -0.016916774213314056,
        0.06637498736381531,
        0.007764773443341255,
        0.0053432355634868145,
        0.017199816182255745
      ]
  }
} 
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment