test.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "from neuromatch.read import read_all_to_all_indexes, read_all_to_all_psame\n",
    "import time\n",
    "import sys\n",
    "from neuromatch.variables import AllToAll, IndexMap\n",
    "from neuromatch.read import read_index_map\n",
    "\n",
    "\n",
    "a = read_all_to_all_indexes(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Cell_reg\\modeled_data_struct.mat\")\n",
    "b = AllToAll(a)\n",
    "b[0, 1, 1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "e:\\Anaconda\\envs\\maze\\lib\\site-packages\\neuromatch\\calculate.py:65: NumbaExperimentalFeatureWarning: \u001b[1m\u001b[1m\u001b[1m\u001b[1m\u001b[1mUse of isinstance() detected. This is an experimental feature.\u001b[0m\u001b[0m\u001b[0m\u001b[0m\u001b[0m\n",
      "  tri_psame = np.triu(psame, k=1)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[  2  33  79  37 115 115  20  30  88   0 185 102   0 214 251 228 208 157\n",
      "   5   0   0  30 125 297  98   0]\n"
     ]
    }
   ],
   "source": [
    "from neuromatch.read import read_all_to_all_indexes, read_all_to_all_psame, read_psame\n",
    "import time\n",
    "import sys\n",
    "import numpy as np\n",
    "from neuromatch.variables import AllToAll, IndexMap\n",
    "from neuromatch.read import read_index_map\n",
    "\n",
    "p_same_ata = read_all_to_all_psame(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref 1\\modeled_data_struct.mat\")\n",
    "p_same = read_psame(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref 1\\cellRegistered.mat\")\n",
    "index_ata = read_all_to_all_indexes(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref 1\\modeled_data_struct.mat\")\n",
    "index_map = read_index_map(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref 1\\cellRegistered.mat\")\n",
    "\n",
    "print(index_map[:, 0])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "1777 1777\n",
      "1777 26 26\n",
      "<class 'neuromatch.variables.alltoall.AllToAll'>\n",
      "26 26 790\n",
      "26 26 790\n",
      "(0, 1) 2->33: p_same 0.63\n",
      "    [0.8597387773236479, 2.6185785218526315e-15, 7.68049845510716e-19, 0.0, 0.0] with IDs [7, 33, 34, 38, 86]\n",
      "(0, 2) 2->79: p_same 0.76\n",
      "    [0.001478132801970229, 0.001927097509612971, 0.0, 0.0, 0.0] with IDs [18, 79, 212, 225, 267]\n",
      "(0, 3) 2->37: p_same 0.42\n",
      "    [0.0, 0.0032471284035022635, 4.497609961639908e-17, 1.6327864628783076e-06, 0.0] with IDs [19, 23, 29, 37, 45]\n",
      "(0, 4) 2->115: p_same nan\n",
      "    [0.0, 0.0, 4.837330227555759e-05, 0.593599575648535, 0.0] with IDs [18, 21, 25, 91, 364]\n",
      "(0, 5) 2->115: p_same nan\n",
      "    [0.001478132801970229, 0.0, 0.11195292908388191, 1.086964406564205e-05, 0.0] with IDs [19, 28, 30, 81, 246]\n",
      "(0, 6) 2->20: p_same nan\n",
      "    [0.0, 0.13849275804139521, 0.0032471284035022635] with IDs [15, 17, 19]\n",
      "(0, 7) 2->30: p_same nan\n",
      "    [0.0, 0.0032471284035022635, 0.3866409643773937, 0.0, 0.0] with IDs [28, 29, 37, 68, 268]\n",
      "(0, 8) 2->88: p_same 0.52\n",
      "    [0.0, 0.0, 0.3866409643773937, 0.0, 0.0] with IDs [32, 38, 52, 88, 149]\n",
      "(0, 9) 2->0: p_same nan\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [3, 57, 134, 287]\n",
      "(0, 10) 2->185: p_same 0.65\n",
      "    [2.2055545282896502e-07, 1.0006686180092675e-19, 0.13849275804139521, 0.0, 1.0006686180092675e-19] with IDs [58, 60, 117, 157, 185]\n",
      "(0, 11) 2->102: p_same nan\n",
      "    [1.994915943251578e-14, 0.08871128330590579, 0.0] with IDs [71, 111, 144]\n",
      "(0, 12) 2->0: p_same nan\n",
      "    [0.0001629137552712127, 8.78197903629907e-12, 2.2055545282896502e-07] with IDs [149, 198, 199]\n",
      "(0, 13) 2->214: p_same 0.44\n",
      "    [0.0, 0.0, 0.0, 0.6805090885369088, 0.0] with IDs [106, 126, 172, 175, 214]\n",
      "(0, 14) 2->251: p_same 0.76\n",
      "    [0.0, 0.0] with IDs [188, 251]\n",
      "(0, 15) 2->228: p_same 0.96\n",
      "    [0.0, 1.994915943251578e-14, 0.4146014322852716, 0.0005021493180715313] with IDs [78, 126, 228, 259]\n",
      "(0, 16) 2->208: p_same 0.99\n",
      "    [0.0, 8.78197903629907e-12, 0.8185267317155368, 0.0] with IDs [66, 173, 208, 316]\n",
      "(0, 17) 2->157: p_same 0.98\n",
      "    [0.0, 7.68049845510716e-19, 0.533121780033363, 0.0, 0.0] with IDs [76, 103, 157, 259, 321]\n",
      "(0, 18) 2->5: p_same 0.56\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [5, 34, 74, 124]\n",
      "(0, 19) 2->0: p_same nan\n",
      "    [6.673279096855399e-11, 4.497609961639908e-17, 0.00688477924895531] with IDs [81, 108, 234]\n",
      "(0, 20) 2->0: p_same nan\n",
      "    [0.0, 0.0, 0.1853063058677883, 0.0] with IDs [52, 84, 109, 382]\n",
      "(0, 21) 2->30: p_same 0.82\n",
      "    [0.0, 0.051301456607682566, 1.1551405927002016e-12, 0.0, 0.0] with IDs [16, 30, 88, 155, 301]\n",
      "(0, 22) 2->125: p_same 0.73\n",
      "    [6.673279096855399e-11, 0.051301456607682566, 4.497609961639908e-17, 0.0001629137552712127, 0.0] with IDs [13, 75, 89, 125, 153]\n",
      "(0, 23) 2->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 386, 420]\n",
      "(0, 24) 2->98: p_same 0.45\n",
      "    [1.086964406564205e-05, 4.497609961639908e-17, 0.0, 0.3337231453401036] with IDs [1, 90, 98, 130]\n",
      "(0, 25) 2->0: p_same nan\n",
      "    [5.068751072752907e-10, 0.5028562145247849, 0.0] with IDs [3, 89, 110]\n",
      "(1, 2) 33->79: p_same 0.99\n",
      "    [0.8486355550032844, 0.0, 0.0, 3.848258134006352e-09] with IDs [79, 102, 225, 267]\n",
      "(1, 3) 33->37: p_same 0.79\n",
      "    [0.3866409643773937, 0.01381338143863724, 0.0] with IDs [29, 37, 45]\n",
      "(1, 4) 33->115: p_same 0.86\n",
      "    [0.0, 0.00688477924895531, 0.13849275804139521, 2.6185785218526315e-15] with IDs [67, 91, 115, 364]\n",
      "(1, 5) 33->115: p_same 0.88\n",
      "    [3.848258134006352e-09, 0.001478132801970229, 0.22180384816300636] with IDs [30, 81, 115]\n",
      "(1, 6) 33->20: p_same 0.49\n",
      "    [0.12479003236933216, 0.0, 0.0] with IDs [17, 20, 75]\n",
      "(1, 7) 33->30: p_same 0.79\n",
      "    [0.00874617914102497, 0.09993833497774224, 0.0] with IDs [29, 30, 68]\n",
      "(1, 8) 33->88: p_same 0.99\n",
      "    [0.0, 0.0006591387107337528, 0.8000704394372794, 0.0] with IDs [32, 52, 88, 147]\n",
      "(1, 9) 33->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [26, 134, 287]\n",
      "(1, 10) 33->185: p_same 0.97\n",
      "    [0.00874617914102497, 1.0006686180092675e-19, 0.0, 0.533121780033363] with IDs [58, 60, 78, 185]\n",
      "(1, 11) 33->102: p_same 0.49\n",
      "    [0.0, 0.22180384816300636, 0.0, 0.051301456607682566] with IDs [47, 71, 102, 111]\n",
      "(1, 12) 33->0: p_same nan\n",
      "    [1.6327864628783076e-06, 0.0, 0.6805090885369088] with IDs [33, 127, 149]\n",
      "(1, 13) 33->214: p_same 0.95\n",
      "    [6.673279096855399e-11, 0.0, 2.6185785218526315e-15, 5.068751072752907e-10, 0.3596480082583055] with IDs [52, 68, 126, 175, 214]\n",
      "(1, 14) 33->251: p_same 0.99\n",
      "    [0.0] with IDs [251]\n",
      "(1, 15) 33->228: p_same 0.45\n",
      "    [1.6327864628783076e-06, 0.6233237033489417, 0.0, 0.0] with IDs [78, 126, 162, 228]\n",
      "(1, 16) 33->208: p_same 0.6\n",
      "    [8.78197903629907e-12, 4.837330227555759e-05, 1.086964406564205e-05] with IDs [66, 129, 208]\n",
      "(1, 17) 33->157: p_same 0.52\n",
      "    [1.5185469713123834e-13, 0.00021977448439905657, 5.882513421515291e-18] with IDs [76, 104, 157]\n",
      "(1, 18) 33->5: p_same 0.97\n",
      "    [0.0, 0.0] with IDs [5, 124]\n",
      "(1, 19) 33->0: p_same nan\n",
      "    [0.0032471284035022635, 2.2055545282896502e-07, 5.882513421515291e-18] with IDs [81, 83, 234]\n",
      "(1, 20) 33->0: p_same nan\n",
      "    [0.00010857896797699931, 0.285342440453865] with IDs [53, 109]\n",
      "(1, 21) 33->30: p_same 0.97\n",
      "    [0.7074996194591217, 0.0] with IDs [30, 155]\n",
      "(1, 22) 33->125: p_same 0.99\n",
      "    [0.0, 3.848258134006352e-09, 0.8000704394372794, 0.0] with IDs [31, 75, 125, 153]\n",
      "(1, 23) 33->297: p_same 0.5\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(1, 24) 33->98: p_same 0.5\n",
      "    [0.11195292908388191, 0.0, 0.051301456607682566, 0.0] with IDs [90, 98, 130, 190]\n",
      "(1, 25) 33->0: p_same nan\n",
      "    [0.0006591387107337528, 0.0, 8.78197903629907e-12] with IDs [89, 110, 159]\n",
      "(2, 3) 79->37: p_same 0.79\n",
      "    [0.0, 0.20299240529798054, 0.07824610650839554, 0.0] with IDs [23, 29, 37, 45]\n",
      "(2, 4) 79->115: p_same 0.79\n",
      "    [0.0, 0.0, 0.22180384816300636, 0.051301456607682566, 0.0] with IDs [21, 67, 91, 115, 364]\n",
      "(2, 5) 79->115: p_same 0.84\n",
      "    [0.004191755897174637, 0.0025056011046007895, 0.12479003236933216] with IDs [30, 81, 115]\n",
      "(2, 6) 79->20: p_same 0.54\n",
      "    [0.5634497139036362, 0.0] with IDs [17, 20]\n",
      "(2, 7) 79->30: p_same 0.73\n",
      "    [0.043796772959800064, 0.07824610650839554, 1.5185469713123834e-13, 2.6185785218526315e-15] with IDs [29, 30, 37, 68]\n",
      "(2, 8) 79->88: p_same 0.95\n",
      "    [0.0, 0.025699798457786676, 0.4146014322852716] with IDs [32, 52, 88]\n",
      "(2, 9) 79->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [3, 134, 287]\n",
      "(2, 10) 79->185: p_same 0.94\n",
      "    [0.12479003236933216, 1.0006686180092675e-19, 0.0, 0.22180384816300636] with IDs [58, 60, 117, 185]\n",
      "(2, 11) 79->102: p_same 0.45\n",
      "    [0.44340020478912745, 0.0, 0.09993833497774224] with IDs [71, 102, 111]\n",
      "(2, 12) 79->0: p_same nan\n",
      "    [1.6327864628783076e-06, 0.0, 0.7572193813866891, 0.0, 0.0] with IDs [33, 127, 149, 198, 199]\n",
      "(2, 13) 79->214: p_same 0.88\n",
      "    [2.2055545282896502e-07, 1.0006686180092675e-19, 0.001927097509612971, 0.11195292908388191] with IDs [52, 126, 175, 214]\n",
      "(2, 14) 79->251: p_same 0.99\n",
      "    [0.0] with IDs [251]\n",
      "(2, 15) 79->228: p_same 0.55\n",
      "    [5.068751072752907e-10, 0.5634497139036362, 3.848258134006352e-09] with IDs [78, 126, 228]\n",
      "(2, 16) 79->208: p_same 0.73\n",
      "    [1.994915943251578e-14, 0.0001629137552712127, 0.011034075838061333] with IDs [66, 129, 208]\n",
      "(2, 17) 79->157: p_same 0.63\n",
      "    [7.68049845510716e-19, 0.0001629137552712127, 0.00010857896797699931] with IDs [76, 104, 157]\n",
      "(2, 18) 79->5: p_same 0.93\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 74, 124]\n",
      "(2, 19) 79->0: p_same nan\n",
      "    [0.0025056011046007895, 6.673279096855399e-11, 2.9189406249663945e-08] with IDs [81, 83, 234]\n",
      "(2, 20) 79->0: p_same nan\n",
      "    [1.086964406564205e-05, 0.0, 0.6805090885369088] with IDs [53, 84, 109]\n",
      "(2, 21) 79->30: p_same 0.99\n",
      "    [0.8707383727915482, 0.0] with IDs [30, 155]\n",
      "(2, 22) 79->125: p_same 0.99\n",
      "    [0.0025056011046007895, 0.8185267317155368, 0.0] with IDs [75, 125, 153]\n",
      "(2, 23) 79->297: p_same 0.54\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(2, 24) 79->98: p_same 0.56\n",
      "    [0.025699798457786676, 0.0, 0.4146014322852716] with IDs [90, 98, 130]\n",
      "(2, 25) 79->0: p_same nan\n",
      "    [0.07824610650839554, 0.0, 7.68049845510716e-19] with IDs [89, 110, 159]\n",
      "(3, 4) 37->115: p_same 0.96\n",
      "    [0.0011312238263153289, 0.004191755897174637, 0.2417816892840939] with IDs [67, 91, 115]\n",
      "(3, 5) 37->115: p_same 0.98\n",
      "    [4.497609961639908e-17, 0.01714717896049333, 0.5028562145247849] with IDs [16, 30, 115]\n",
      "(3, 6) 37->20: p_same 0.83\n",
      "    [0.3337231453401036, 2.9189406249663945e-08, 0.0002904091994534483] with IDs [17, 18, 20]\n",
      "(3, 7) 37->30: p_same 0.96\n",
      "    [3.848258134006352e-09, 0.533121780033363, 0.001927097509612971] with IDs [9, 30, 68]\n",
      "(3, 8) 37->88: p_same 0.84\n",
      "    [0.0002904091994534483, 8.78197903629907e-12, 0.01381338143863724] with IDs [27, 32, 88]\n",
      "(3, 9) 37->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [26, 287]\n",
      "(3, 10) 37->185: p_same 0.58\n",
      "    [0.00874617914102497, 0.285342440453865, 0.0, 1.994915943251578e-14, 0.0] with IDs [55, 58, 71, 185, 329]\n",
      "(3, 11) 37->102: p_same 0.81\n",
      "    [0.3866409643773937, 0.13849275804139521, 0.0] with IDs [71, 102, 153]\n",
      "(3, 12) 37->0: p_same nan\n",
      "    [0.652374076473276, 0.00038214183730195733] with IDs [33, 149]\n",
      "(3, 13) 37->214: p_same 0.86\n",
      "    [0.4146014322852716, 0.0025056011046007895] with IDs [52, 214]\n",
      "(3, 14) 37->251: p_same 0.74\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(3, 15) 37->228: p_same nan\n",
      "    [0.3337231453401036, 0.0002904091994534483, 0.0] with IDs [126, 162, 262]\n",
      "(3, 16) 37->208: p_same 0.42\n",
      "    [0.733133746520555, 0.00010857896797699931, 1.0006686180092675e-19] with IDs [129, 208, 209]\n",
      "(3, 17) 37->157: p_same nan\n",
      "    [0.6233237033489417, 3.848258134006352e-09] with IDs [104, 212]\n",
      "(3, 18) 37->5: p_same 0.61\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(3, 19) 37->0: p_same nan\n",
      "    [0.22180384816300636, 0.0001629137552712127] with IDs [83, 234]\n",
      "(3, 20) 37->0: p_same nan\n",
      "    [0.2417816892840939, 0.037032562369221755] with IDs [53, 109]\n",
      "(3, 21) 37->30: p_same 0.74\n",
      "    [0.043796772959800064, 3.848258134006352e-09, 5.068751072752907e-10] with IDs [30, 54, 240]\n",
      "(3, 22) 37->125: p_same 0.76\n",
      "    [1.994915943251578e-14, 0.0, 0.00874617914102497, 0.00688477924895531] with IDs [17, 65, 75, 125]\n",
      "(3, 23) 37->297: p_same 0.86\n",
      "    [0.0] with IDs [297]\n",
      "(3, 24) 37->98: p_same 0.79\n",
      "    [0.0, 4.837330227555759e-05, 0.021093077719105825, 0.0025056011046007895] with IDs [90, 98, 130, 179]\n",
      "(3, 25) 37->0: p_same nan\n",
      "    [0.0032471284035022635, 2.6185785218526315e-15, 2.9189406249663945e-08] with IDs [89, 110, 159]\n",
      "(4, 5) 115->115: p_same 0.99\n",
      "    [0.0, 4.497609961639908e-17, 0.0, 0.0, 0.8485634231160908] with IDs [16, 30, 42, 81, 115]\n",
      "(4, 6) 115->20: p_same 0.58\n",
      "    [0.025699798457786676, 0.0, 1.0006686180092675e-19, 0.0] with IDs [17, 18, 20, 75]\n",
      "(4, 7) 115->30: p_same 0.99\n",
      "    [0.0, 0.8000704394372794, 0.0, 0.0] with IDs [9, 30, 64, 68]\n",
      "(4, 8) 115->88: p_same 0.94\n",
      "    [1.994915943251578e-14, 0.0, 3.4338416717859466e-16, 0.44340020478912745, 0.0] with IDs [27, 32, 69, 88, 147]\n",
      "(4, 9) 115->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [26, 287]\n",
      "(4, 10) 115->185: p_same 0.67\n",
      "    [0.0008641445255239377, 0.0032471284035022635, 0.0, 0.0001629137552712127] with IDs [55, 58, 78, 185]\n",
      "(4, 11) 115->102: p_same 0.86\n",
      "    [0.0, 0.12479003236933216, 0.025699798457786676] with IDs [47, 71, 102]\n",
      "(4, 12) 115->0: p_same nan\n",
      "    [0.3337231453401036, 0.0, 0.0005021493180715313] with IDs [33, 127, 149]\n",
      "(4, 13) 115->214: p_same 0.97\n",
      "    [0.00874617914102497, 0.652374076473276] with IDs [52, 214]\n",
      "(4, 14) 115->251: p_same 0.76\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(4, 15) 115->228: p_same nan\n",
      "    [0.733133746520555, 0.00874617914102497] with IDs [126, 162]\n",
      "(4, 16) 115->208: p_same nan\n",
      "    [0.13849275804139521, 8.78197903629907e-12] with IDs [129, 209]\n",
      "(4, 17) 115->157: p_same nan\n",
      "    [0.44340020478912745] with IDs [104]\n",
      "(4, 18) 115->5: p_same 0.74\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(4, 19) 115->0: p_same nan\n",
      "    [0.3596480082583055] with IDs [83]\n",
      "(4, 20) 115->0: p_same nan\n",
      "    [0.593599575648535, 0.001478132801970229, 0.0] with IDs [53, 109, 309]\n",
      "(4, 21) 115->30: p_same 0.72\n",
      "    [0.021093077719105825, 0.00688477924895531] with IDs [30, 54]\n",
      "(4, 22) 115->125: p_same 0.79\n",
      "    [3.848258134006352e-09, 5.882513421515291e-18, 0.09993833497774224] with IDs [65, 75, 125]\n",
      "(4, 23) 115->297: p_same 0.62\n",
      "    [0.0, 0.0, 0.0] with IDs [19, 209, 297]\n",
      "(4, 24) 115->98: p_same 0.55\n",
      "    [1.5185469713123834e-13, 0.0, 2.2055545282896502e-07, 0.01714717896049333, 0.0] with IDs [90, 98, 130, 179, 190]\n",
      "(4, 25) 115->0: p_same nan\n",
      "    [4.497609961639908e-17, 0.0, 0.06852666906345631] with IDs [89, 110, 159]\n",
      "(5, 6) 115->20: p_same 0.65\n",
      "    [0.08871128330590579, 0.0, 1.994915943251578e-14, 0.0] with IDs [17, 18, 20, 75]\n",
      "(5, 7) 115->30: p_same 0.99\n",
      "    [0.0, 0.8347869602414921, 4.497609961639908e-17] with IDs [9, 30, 68]\n",
      "(5, 8) 115->88: p_same 0.94\n",
      "    [2.6185785218526315e-15, 0.0, 0.0, 0.0, 0.47287981784443234] with IDs [27, 32, 52, 69, 88]\n",
      "(5, 9) 115->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [26, 287]\n",
      "(5, 10) 115->185: p_same 0.68\n",
      "    [0.0005021493180715313, 0.03100466054425117, 0.0, 0.0002904091994534483] with IDs [55, 58, 78, 185]\n",
      "(5, 11) 115->102: p_same 0.83\n",
      "    [0.0, 0.285342440453865, 0.025699798457786676] with IDs [47, 71, 102]\n",
      "(5, 12) 115->0: p_same nan\n",
      "    [0.4146014322852716, 0.0, 0.001927097509612971] with IDs [33, 127, 149]\n",
      "(5, 13) 115->214: p_same 0.96\n",
      "    [0.043796772959800064, 0.44340020478912745] with IDs [52, 214]\n",
      "(5, 14) 115->251: p_same 0.8\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(5, 15) 115->228: p_same nan\n",
      "    [0.8185267317155368, 0.0011312238263153289] with IDs [126, 162]\n",
      "(5, 16) 115->208: p_same nan\n",
      "    [0.285342440453865, 7.68049845510716e-19] with IDs [129, 209]\n",
      "(5, 17) 115->157: p_same nan\n",
      "    [0.533121780033363, 0.0] with IDs [104, 212]\n",
      "(5, 18) 115->5: p_same 0.74\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(5, 19) 115->0: p_same nan\n",
      "    [0.262955983487866] with IDs [83]\n",
      "(5, 20) 115->0: p_same nan\n",
      "    [0.5634497139036362, 0.00688477924895531, 0.0] with IDs [53, 109, 309]\n",
      "(5, 21) 115->30: p_same 0.77\n",
      "    [0.051301456607682566, 0.0005021493180715313, 0.0] with IDs [30, 54, 240]\n",
      "(5, 22) 115->125: p_same 0.83\n",
      "    [0.0, 4.497609961639908e-17, 1.5185469713123834e-13, 0.15311053170813824] with IDs [17, 65, 75, 125]\n",
      "(5, 23) 115->297: p_same 0.69\n",
      "    [0.0, 0.0, 0.0] with IDs [19, 209, 297]\n",
      "(5, 24) 115->98: p_same 0.62\n",
      "    [1.994915943251578e-14, 1.0006686180092675e-19, 0.0002904091994534483, 0.005386434354395895] with IDs [90, 98, 130, 179]\n",
      "(5, 25) 115->0: p_same nan\n",
      "    [1.1551405927002016e-12, 0.0, 0.00688477924895531] with IDs [89, 110, 159]\n",
      "(6, 7) 20->30: p_same 0.59\n",
      "    [3.4338416717859466e-16, 0.15311053170813824, 5.882513421515291e-18] with IDs [30, 68, 74]\n",
      "(6, 8) 20->88: p_same 0.48\n",
      "    [0.533121780033363, 0.0, 0.0] with IDs [32, 83, 88]\n",
      "(6, 9) 20->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [78, 287]\n",
      "(6, 10) 20->185: p_same nan\n",
      "    [4.837330227555759e-05, 7.68049845510716e-19, 0.07824610650839554] with IDs [58, 71, 329]\n",
      "(6, 11) 20->102: p_same 0.52\n",
      "    [1.994915943251578e-14, 1.1551405927002016e-12, 2.6185785218526315e-15, 2.9189406249663945e-08] with IDs [71, 102, 153, 281]\n",
      "(6, 12) 20->0: p_same nan\n",
      "    [0.0025056011046007895, 1.1551405927002016e-12, 0.0] with IDs [33, 111, 199]\n",
      "(6, 13) 20->214: p_same 0.47\n",
      "    [0.262955983487866, 0.0] with IDs [52, 214]\n",
      "(6, 14) 20->251: p_same 0.49\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(6, 15) 20->228: p_same nan\n",
      "    [1.0006686180092675e-19, 0.004191755897174637] with IDs [126, 262]\n",
      "(6, 16) 20->208: p_same nan\n",
      "    [0.01714717896049333, 0.0011312238263153289] with IDs [129, 195]\n",
      "(6, 17) 20->157: p_same nan\n",
      "    [2.2055545282896502e-07, 0.01381338143863724] with IDs [104, 212]\n",
      "(6, 18) 20->5: p_same nan\n",
      "    [0.0, 0.0] with IDs [9, 124]\n",
      "(6, 19) 20->0: p_same nan\n",
      "    [1.086964406564205e-05, 0.0001629137552712127] with IDs [83, 234]\n",
      "(6, 20) 20->0: p_same nan\n",
      "    [2.6185785218526315e-15, 0.0, 0.07824610650839554] with IDs [53, 109, 240]\n",
      "(6, 21) 20->30: p_same 0.52\n",
      "    [0.0, 0.285342440453865] with IDs [30, 240]\n",
      "(6, 22) 20->125: p_same 0.49\n",
      "    [0.20299240529798054, 0.0, 0.0] with IDs [17, 75, 125]\n",
      "(6, 23) 20->297: p_same 0.99\n",
      "    [0.0] with IDs [297]\n",
      "(6, 24) 20->98: p_same 0.99\n",
      "    [0.8597387773236479, 0.0] with IDs [98, 130]\n",
      "(6, 25) 20->0: p_same nan\n",
      "    [1.0006686180092675e-19, 0.47287981784443234] with IDs [89, 110]\n",
      "(7, 8) 30->88: p_same 0.89\n",
      "    [0.0032471284035022635, 0.0, 2.9189406249663945e-08, 0.20299240529798054] with IDs [27, 32, 69, 88]\n",
      "(7, 9) 30->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [26, 287]\n",
      "(7, 10) 30->185: p_same 0.6\n",
      "    [0.11195292908388191, 0.037032562369221755, 0.0, 1.086964406564205e-05] with IDs [55, 58, 78, 185]\n",
      "(7, 11) 30->102: p_same 0.91\n",
      "    [0.0, 0.22180384816300636, 0.22180384816300636] with IDs [47, 71, 102]\n",
      "(7, 12) 30->0: p_same nan\n",
      "    [0.5634497139036362, 0.0, 0.0008641445255239377] with IDs [33, 127, 149]\n",
      "(7, 13) 30->214: p_same 0.94\n",
      "    [0.043796772959800064, 0.22180384816300636] with IDs [52, 214]\n",
      "(7, 14) 30->251: p_same 0.69\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(7, 15) 30->228: p_same nan\n",
      "    [0.6233237033489417, 0.1853063058677883] with IDs [126, 162]\n",
      "(7, 16) 30->208: p_same nan\n",
      "    [0.3089391001392501, 0.005386434354395895] with IDs [129, 209]\n",
      "(7, 17) 30->157: p_same nan\n",
      "    [0.7074996194591217, 0.0] with IDs [104, 212]\n",
      "(7, 18) 30->5: p_same 0.66\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(7, 19) 30->0: p_same nan\n",
      "    [0.6233237033489417] with IDs [83]\n",
      "(7, 20) 30->0: p_same nan\n",
      "    [0.7074996194591217, 0.005386434354395895, 0.0] with IDs [53, 109, 309]\n",
      "(7, 21) 30->30: p_same 0.66\n",
      "    [0.025699798457786676, 0.08871128330590579] with IDs [30, 54]\n",
      "(7, 22) 30->125: p_same 0.72\n",
      "    [0.0, 0.001927097509612971, 5.068751072752907e-10, 0.06852666906345631] with IDs [17, 65, 75, 125]\n",
      "(7, 23) 30->297: p_same 0.64\n",
      "    [0.0, 0.0, 0.0] with IDs [19, 209, 297]\n",
      "(7, 24) 30->98: p_same 0.55\n",
      "    [4.497609961639908e-17, 1.0006686180092675e-19, 0.00038214183730195733, 0.16869678193966425, 0.0] with IDs [90, 98, 130, 179, 190]\n",
      "(7, 25) 30->0: p_same nan\n",
      "    [5.068751072752907e-10, 0.07824610650839554] with IDs [89, 159]\n",
      "(8, 9) 88->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [26, 134, 287]\n",
      "(8, 10) 88->185: p_same 0.93\n",
      "    [0.0, 0.001927097509612971, 0.0, 0.0, 0.3866409643773937] with IDs [55, 58, 60, 78, 185]\n",
      "(8, 11) 88->102: p_same 0.58\n",
      "    [0.0, 0.12479003236933216, 3.4338416717859466e-16, 0.0025056011046007895] with IDs [47, 71, 102, 111]\n",
      "(8, 12) 88->0: p_same nan\n",
      "    [0.001927097509612971, 0.0, 0.1853063058677883] with IDs [33, 127, 149]\n",
      "(8, 13) 88->214: p_same 0.99\n",
      "    [1.086964406564205e-05, 0.0, 2.6185785218526315e-15, 0.0, 0.6805090885369088] with IDs [52, 68, 126, 175, 214]\n",
      "(8, 14) 88->251: p_same 0.95\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(8, 15) 88->228: p_same nan\n",
      "    [5.068751072752907e-10, 0.6233237033489417, 4.497609961639908e-17] with IDs [78, 126, 162]\n",
      "(8, 16) 88->208: p_same 0.49\n",
      "    [1.994915943251578e-14, 0.0008641445255239377, 7.68049845510716e-19] with IDs [66, 129, 208]\n",
      "(8, 17) 88->157: p_same 0.43\n",
      "    [1.5185469713123834e-13, 0.005386434354395895, 0.0] with IDs [76, 104, 157]\n",
      "(8, 18) 88->5: p_same 0.96\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(8, 19) 88->0: p_same nan\n",
      "    [0.00038214183730195733, 0.001478132801970229] with IDs [81, 83]\n",
      "(8, 20) 88->0: p_same nan\n",
      "    [0.00874617914102497, 0.037032562369221755, 0.0] with IDs [53, 109, 309]\n",
      "(8, 21) 88->30: p_same 0.92\n",
      "    [0.2417816892840939, 3.4338416717859466e-16, 0.0] with IDs [30, 54, 155]\n",
      "(8, 22) 88->125: p_same 0.96\n",
      "    [0.0, 0.0, 5.882513421515291e-18, 0.533121780033363] with IDs [31, 65, 75, 125]\n",
      "(8, 23) 88->297: p_same 0.5\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(8, 24) 88->98: p_same 0.48\n",
      "    [0.06852666906345631, 0.0, 0.00038214183730195733, 0.0] with IDs [90, 98, 130, 190]\n",
      "(8, 25) 88->0: p_same nan\n",
      "    [1.1551405927002016e-12, 0.0001629137552712127] with IDs [89, 159]\n",
      "(9, 10) 0->185: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [290, 420, 584]\n",
      "(9, 11) 0->102: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [167, 226, 565]\n",
      "(9, 12) 0->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [194, 456, 593]\n",
      "(9, 13) 0->214: p_same nan\n",
      "    [0.0, 0.0] with IDs [264, 316]\n",
      "(9, 14) 0->251: p_same nan\n",
      "    [0.0] with IDs [431]\n",
      "(9, 15) 0->228: p_same nan\n",
      "    [0.0] with IDs [166]\n",
      "(9, 16) 0->208: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [258, 298, 543]\n",
      "(9, 17) 0->157: p_same nan\n",
      "    [0.0] with IDs [397]\n",
      "(9, 18) 0->5: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(9, 19) 0->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [299, 616]\n",
      "(9, 20) 0->0: p_same nan\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [527, 641, 802, 811]\n",
      "(9, 21) 0->30: p_same nan\n",
      "    [0.0] with IDs [540]\n",
      "(9, 22) 0->125: p_same nan\n",
      "    [0.0, 0.0] with IDs [677, 687]\n",
      "(9, 23) 0->297: p_same nan\n",
      "    [0.0] with IDs [578]\n",
      "(9, 24) 0->98: p_same nan\n",
      "    [0.0, 0.0] with IDs [315, 488]\n",
      "(9, 25) 0->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [181, 511]\n",
      "(10, 11) 185->102: p_same nan\n",
      "    [0.0, 4.837330227555759e-05, 0.15311053170813824] with IDs [47, 71, 111]\n",
      "(10, 12) 185->0: p_same nan\n",
      "    [0.0, 0.0, 0.6805090885369088, 0.0, 7.68049845510716e-19] with IDs [33, 127, 149, 198, 357]\n",
      "(10, 13) 185->214: p_same 0.85\n",
      "    [0.0, 0.0, 0.004191755897174637, 8.78197903629907e-12, 0.025699798457786676] with IDs [52, 68, 126, 175, 214]\n",
      "(10, 14) 185->251: p_same 0.97\n",
      "    [0.0, 0.0] with IDs [191, 251]\n",
      "(10, 15) 185->228: p_same 0.45\n",
      "    [0.03100466054425117, 0.005386434354395895, 0.0] with IDs [78, 126, 228]\n",
      "(10, 16) 185->208: p_same 0.6\n",
      "    [0.005386434354395895, 0.0, 8.78197903629907e-12] with IDs [66, 129, 208]\n",
      "(10, 17) 185->157: p_same 0.5\n",
      "    [0.00688477924895531, 0.0, 0.0] with IDs [76, 104, 157]\n",
      "(10, 18) 185->5: p_same 0.99\n",
      "    [0.0] with IDs [5]\n",
      "(10, 19) 185->0: p_same nan\n",
      "    [0.13849275804139521, 0.0, 0.0] with IDs [81, 83, 204]\n",
      "(10, 20) 185->0: p_same nan\n",
      "    [0.0, 1.1551405927002016e-12, 0.021093077719105825] with IDs [53, 84, 109]\n",
      "(10, 21) 185->30: p_same 0.94\n",
      "    [0.1853063058677883, 0.0, 8.78197903629907e-12] with IDs [30, 69, 155]\n",
      "(10, 22) 185->125: p_same 0.97\n",
      "    [0.0, 0.0, 0.3866409643773937, 2.2055545282896502e-07] with IDs [31, 75, 125, 153]\n",
      "(10, 23) 185->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 209, 386]\n",
      "(10, 24) 185->98: p_same nan\n",
      "    [0.652374076473276, 0.0008641445255239377, 0.0] with IDs [90, 130, 190]\n",
      "(10, 25) 185->0: p_same nan\n",
      "    [2.9189406249663945e-08, 0.0] with IDs [89, 204]\n",
      "(11, 12) 102->0: p_same nan\n",
      "    [0.652374076473276] with IDs [33]\n",
      "(11, 13) 102->214: p_same 0.67\n",
      "    [0.043796772959800064, 1.5185469713123834e-13] with IDs [52, 214]\n",
      "(11, 14) 102->251: p_same nan\n",
      "    [0.0, 0.0] with IDs [3, 415]\n",
      "(11, 15) 102->228: p_same nan\n",
      "    [3.848258134006352e-09, 0.3337231453401036] with IDs [126, 162]\n",
      "(11, 16) 102->208: p_same nan\n",
      "    [0.285342440453865, 0.0032471284035022635] with IDs [129, 209]\n",
      "(11, 17) 102->157: p_same nan\n",
      "    [0.733133746520555, 1.1551405927002016e-12] with IDs [104, 212]\n",
      "(11, 18) 102->5: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [9, 95, 122]\n",
      "(11, 19) 102->0: p_same nan\n",
      "    [0.6233237033489417] with IDs [83]\n",
      "(11, 20) 102->0: p_same nan\n",
      "    [0.4146014322852716] with IDs [53]\n",
      "(11, 21) 102->30: p_same nan\n",
      "    [0.05954546342307142, 7.68049845510716e-19, 1.5185469713123834e-13] with IDs [54, 240, 336]\n",
      "(11, 22) 102->125: p_same 0.44\n",
      "    [7.68049845510716e-19, 0.00038214183730195733, 5.882513421515291e-18] with IDs [17, 65, 125]\n",
      "(11, 23) 102->297: p_same 0.58\n",
      "    [0.0, 0.0] with IDs [19, 297]\n",
      "(11, 24) 102->98: p_same 0.46\n",
      "    [4.497609961639908e-17, 0.262955983487866] with IDs [98, 179]\n",
      "(11, 25) 102->0: p_same nan\n",
      "    [0.001927097509612971] with IDs [159]\n",
      "(12, 13) 0->214: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 14) 0->251: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 15) 0->228: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 16) 0->208: p_same nan\n",
      "    [0.043796772959800064] with IDs [553]\n",
      "(12, 17) 0->157: p_same nan\n",
      "    [0.1853063058677883] with IDs [629]\n",
      "(12, 18) 0->5: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 19) 0->0: p_same nan\n",
      "    [0.011034075838061333] with IDs [633]\n",
      "(12, 20) 0->0: p_same nan\n",
      "    [0.025699798457786676] with IDs [848]\n",
      "(12, 21) 0->30: p_same nan\n",
      "    [3.4338416717859466e-16] with IDs [694]\n",
      "(12, 22) 0->125: p_same nan\n",
      "    [0.03100466054425117] with IDs [710]\n",
      "(12, 23) 0->297: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 24) 0->98: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(12, 25) 0->0: p_same nan\n",
      "    [0.0025056011046007895] with IDs [584]\n",
      "(13, 14) 214->251: p_same 0.88\n",
      "    [0.0, 0.0] with IDs [251, 415]\n",
      "(13, 15) 214->228: p_same nan\n",
      "    [0.0, 0.5028562145247849, 4.837330227555759e-05] with IDs [78, 126, 162]\n",
      "(13, 16) 214->208: p_same nan\n",
      "    [0.0, 0.0, 4.837330227555759e-05, 2.6185785218526315e-15] with IDs [66, 105, 129, 209]\n",
      "(13, 17) 214->157: p_same nan\n",
      "    [0.0, 0.005386434354395895] with IDs [76, 104]\n",
      "(13, 18) 214->5: p_same 0.91\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 9, 124]\n",
      "(13, 19) 214->0: p_same nan\n",
      "    [3.4338416717859466e-16, 0.01714717896049333] with IDs [81, 83]\n",
      "(13, 20) 214->0: p_same nan\n",
      "    [0.01714717896049333, 0.00874617914102497, 0.0] with IDs [53, 109, 309]\n",
      "(13, 21) 214->30: p_same 0.83\n",
      "    [0.09993833497774224, 0.0005021493180715313] with IDs [30, 54]\n",
      "(13, 22) 214->125: p_same 0.9\n",
      "    [0.0, 2.9189406249663945e-08, 0.0, 0.1853063058677883] with IDs [31, 65, 75, 125]\n",
      "(13, 23) 214->297: p_same 0.5\n",
      "    [0.0, 0.0, 0.0] with IDs [209, 297, 386]\n",
      "(13, 24) 214->98: p_same 0.46\n",
      "    [0.0006591387107337528, 0.0, 0.00010857896797699931, 0.0002904091994534483, 0.0] with IDs [90, 98, 130, 179, 190]\n",
      "(13, 25) 214->0: p_same nan\n",
      "    [4.497609961639908e-17, 0.03100466054425117] with IDs [89, 159]\n",
      "(14, 15) 251->228: p_same 0.55\n",
      "    [0.0, 0.0, 0.0] with IDs [78, 126, 228]\n",
      "(14, 16) 251->208: p_same 0.73\n",
      "    [0.0, 0.0, 0.0] with IDs [66, 129, 208]\n",
      "(14, 17) 251->157: p_same 0.63\n",
      "    [0.0, 0.0, 0.0] with IDs [76, 104, 157]\n",
      "(14, 18) 251->5: p_same 0.95\n",
      "    [0.0, 0.0, 0.0] with IDs [5, 74, 124]\n",
      "(14, 19) 251->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [81, 83, 234]\n",
      "(14, 20) 251->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [53, 84, 109]\n",
      "(14, 21) 251->30: p_same 0.99\n",
      "    [0.0, 0.0] with IDs [30, 155]\n",
      "(14, 22) 251->125: p_same 0.99\n",
      "    [0.0, 0.0, 0.0] with IDs [75, 125, 153]\n",
      "(14, 23) 251->297: p_same 0.49\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(14, 24) 251->98: p_same 0.51\n",
      "    [0.0, 0.0, 0.0] with IDs [90, 98, 130]\n",
      "(14, 25) 251->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [89, 110]\n",
      "(15, 16) 228->208: p_same 0.97\n",
      "    [0.0, 1.1551405927002016e-12, 0.7795840112233621, 0.00010857896797699931] with IDs [66, 173, 208, 316]\n",
      "(15, 17) 228->157: p_same 0.99\n",
      "    [0.0, 1.5185469713123834e-13, 0.8675020841295138, 0.0, 2.2055545282896502e-07] with IDs [76, 103, 157, 259, 321]\n",
      "(15, 18) 228->5: p_same nan\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [34, 74, 124, 183]\n",
      "(15, 19) 228->0: p_same nan\n",
      "    [0.0, 2.2055545282896502e-07, 0.22180384816300636] with IDs [81, 108, 234]\n",
      "(15, 20) 228->0: p_same nan\n",
      "    [7.68049845510716e-19, 0.0, 0.051301456607682566, 5.068751072752907e-10] with IDs [52, 84, 109, 382]\n",
      "(15, 21) 228->30: p_same 0.6\n",
      "    [0.0, 0.001478132801970229, 0.011034075838061333, 3.4338416717859466e-16] with IDs [16, 30, 88, 301]\n",
      "(15, 22) 228->125: p_same 0.52\n",
      "    [4.497609961639908e-17, 0.11195292908388191, 0.004191755897174637, 3.4338416717859466e-16, 1.0006686180092675e-19] with IDs [13, 75, 89, 125, 128]\n",
      "(15, 23) 228->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 386, 420]\n",
      "(15, 24) 228->98: p_same nan\n",
      "    [0.0002904091994534483, 2.2055545282896502e-07, 0.0, 0.11195292908388191] with IDs [1, 69, 90, 130]\n",
      "(15, 25) 228->0: p_same nan\n",
      "    [5.068751072752907e-10, 0.3089391001392501, 0.0] with IDs [3, 89, 110]\n",
      "(16, 17) 208->157: p_same 0.99\n",
      "    [0.0, 3.4338416717859466e-16, 0.8347869602414921, 0.0] with IDs [76, 103, 157, 321]\n",
      "(16, 18) 208->5: p_same 0.53\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [5, 34, 74, 124]\n",
      "(16, 19) 208->0: p_same nan\n",
      "    [6.673279096855399e-11, 1.1551405927002016e-12, 0.06852666906345631] with IDs [81, 108, 234]\n",
      "(16, 20) 208->0: p_same nan\n",
      "    [0.0, 0.0, 0.3089391001392501, 0.0] with IDs [52, 84, 109, 382]\n",
      "(16, 21) 208->30: p_same 0.79\n",
      "    [0.0, 0.11195292908388191, 0.00021977448439905657, 0.0, 0.0] with IDs [16, 30, 88, 155, 301]\n",
      "(16, 22) 208->125: p_same 0.69\n",
      "    [1.1551405927002016e-12, 0.13849275804139521, 1.6327864628783076e-06, 0.0011312238263153289, 0.0] with IDs [13, 75, 89, 125, 153]\n",
      "(16, 23) 208->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 386, 420]\n",
      "(16, 24) 208->98: p_same 0.46\n",
      "    [0.00010857896797699931, 2.6185785218526315e-15, 0.0, 0.5634497139036362] with IDs [1, 90, 98, 130]\n",
      "(16, 25) 208->0: p_same nan\n",
      "    [5.068751072752907e-10, 0.6805090885369088, 0.0] with IDs [3, 89, 110]\n",
      "(17, 18) 157->5: p_same 0.44\n",
      "    [0.0, 0.0, 0.0, 0.0, 0.0] with IDs [5, 34, 74, 124, 183]\n",
      "(17, 19) 157->0: p_same nan\n",
      "    [1.0006686180092675e-19, 1.994915943251578e-14, 0.22180384816300636] with IDs [81, 108, 234]\n",
      "(17, 20) 157->0: p_same nan\n",
      "    [0.0, 0.0, 0.13849275804139521, 7.68049845510716e-19] with IDs [52, 84, 109, 382]\n",
      "(17, 21) 157->30: p_same 0.69\n",
      "    [0.0, 0.00874617914102497, 0.0005021493180715313, 0.0] with IDs [16, 30, 88, 301]\n",
      "(17, 22) 157->125: p_same 0.59\n",
      "    [0.0, 0.2417816892840939, 0.0001629137552712127, 1.1551405927002016e-12, 0.0] with IDs [13, 75, 89, 125, 128]\n",
      "(17, 23) 157->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 386, 420]\n",
      "(17, 24) 157->98: p_same 0.47\n",
      "    [3.848258134006352e-09, 0.0, 0.0, 0.2417816892840939] with IDs [1, 90, 98, 130]\n",
      "(17, 25) 157->0: p_same nan\n",
      "    [3.4338416717859466e-16, 0.5634497139036362, 0.0] with IDs [3, 89, 110]\n",
      "(18, 19) 5->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [81, 83, 204]\n",
      "(18, 20) 5->0: p_same nan\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [53, 84, 109, 309]\n",
      "(18, 21) 5->30: p_same 0.91\n",
      "    [0.0, 0.0, 0.0] with IDs [30, 69, 155]\n",
      "(18, 22) 5->125: p_same 0.96\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [31, 75, 125, 153]\n",
      "(18, 23) 5->297: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [50, 209, 386]\n",
      "(18, 24) 5->98: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [90, 130, 190]\n",
      "(18, 25) 5->0: p_same nan\n",
      "    [0.0, 0.0, 0.0] with IDs [89, 159, 204]\n",
      "(19, 20) 0->0: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(19, 21) 0->30: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(19, 22) 0->125: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(19, 23) 0->297: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(19, 24) 0->98: p_same nan\n",
      "    [0.6805090885369088] with IDs [579]\n",
      "(19, 25) 0->0: p_same nan\n",
      "    [0.6233237033489417] with IDs [571]\n",
      "(20, 21) 0->30: p_same nan\n",
      "    [0.00010857896797699931, 4.497609961639908e-17] with IDs [653, 687]\n",
      "(20, 22) 0->125: p_same nan\n",
      "    [0.00010857896797699931] with IDs [646]\n",
      "(20, 23) 0->297: p_same nan\n",
      "    [0.0, 0.0] with IDs [96, 318]\n",
      "(20, 24) 0->98: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(20, 25) 0->0: p_same nan\n",
      "    [0, 0] with IDs [0, 0]\n",
      "(21, 22) 30->125: p_same 0.99\n",
      "    [0.05954546342307142, 0.7074996194591217, 0.0] with IDs [75, 125, 153]\n",
      "(21, 23) 30->297: p_same 0.52\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(21, 24) 30->98: p_same 0.55\n",
      "    [0.025699798457786676, 0.0, 0.733133746520555] with IDs [90, 98, 130]\n",
      "(21, 25) 30->0: p_same nan\n",
      "    [0.262955983487866, 0.0] with IDs [89, 110]\n",
      "(22, 23) 125->297: p_same 0.5\n",
      "    [0.0, 0.0, 0.0, 0.0] with IDs [50, 209, 297, 386]\n",
      "(22, 24) 125->98: p_same 0.51\n",
      "    [0.13849275804139521, 0.0, 0.08871128330590579] with IDs [90, 98, 130]\n",
      "(22, 25) 125->0: p_same nan\n",
      "    [0.004191755897174637, 0.0, 1.5185469713123834e-13] with IDs [89, 110, 159]\n",
      "(23, 24) 297->98: p_same 0.98\n",
      "    [0.0, 0.0] with IDs [98, 130]\n",
      "(23, 25) 297->0: p_same nan\n",
      "    [0.0, 0.0] with IDs [89, 110]\n",
      "(24, 25) 98->0: p_same nan\n",
      "    [1.5185469713123834e-13, 0.733133746520555] with IDs [89, 110]\n"
     ]
    }
   ],
   "source": [
    "print(len(p_same), index_map.shape[1])\n",
    "print(len(p_same), len(p_same[0]), len(p_same[0, 0]))\n",
    "print(type(p_same_ata))\n",
    "print(len(p_same_ata), len(p_same_ata[0]), len(p_same_ata[0, 0]))\n",
    "print(len(index_ata), len(index_ata[0]), len(index_ata[0, 0]))\n",
    "for i in range(index_map[:, 0].shape[0]-1):\n",
    "    for j in range(i+1, index_map[:, 0].shape[0]):\n",
    "        print(f\"{i, j} {index_map[i, 0]}->{index_map[j, 0]}: p_same {round(p_same[0, i, j], 2)}\")\n",
    "        print(f\"    {p_same_ata[i, j, index_map[i, 0]-1]} with IDs {index_ata[i, j, index_map[i, 0]-1]}\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[  3   0 122   0  28  61  16  50   6   1   7  13 175 202 266 274 165 326\n",
      "   0 198 186 395 259 175   0 182]\n"
     ]
    }
   ],
   "source": [
    "from neuromatch.read import read_index_map, read_all_to_all_psame, read_all_to_all_indexes\n",
    "from neuromatch.variables.regneuron import RegisteredNeuron\n",
    "import pickle\n",
    "    \n",
    "model_dir = r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Cell_reg\\modeled_data_struct.mat\"\n",
    "index_map = read_index_map(r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Cell_reg\\cellRegistered.mat\")\n",
    "    \n",
    "print(index_map[:, 2])\n",
    "    \n",
    "ref_dirs = [\n",
    "    r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref \"+str(i)+r\"\\cellRegistered.mat\" for i in [1, 4, 7, 10, 15, 17, 20, 23, 26]\n",
    "]\n",
    "ref_model_dir = [\n",
    "    r\"E:\\Data\\Cross_maze\\10227\\Super Long-term Maze 1\\Ref \"+str(i)+r\"\\modeled_data_struct.mat\" for i in [1, 4, 7, 10, 15, 17, 20, 23, 26]\n",
    "]\n",
    "\n",
    "ref_indexmaps = [read_index_map(dir_name) for dir_name in ref_dirs]\n",
    "ata_p_sames = [read_all_to_all_psame(dir_name) for dir_name in [model_dir]+ref_model_dir]\n",
    "ata_indexmaps = [read_all_to_all_indexes(dir_name) for dir_name in [model_dir]+ref_model_dir]\n",
    "with open(r\"E:\\Anaconda\\envs\\maze\\Lib\\site-packages\\neuromatch\\temp.pkl\", 'wb') as f:\n",
    "    pickle.dump([index_map, ref_indexmaps, ata_p_sames, ata_indexmaps], f)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      " 1. Done initialization.\n",
      " 2. All candidates were found.\n",
      " 3. Done vectorization of candidates.\n",
      " 4. Done vectorization of psame matrix\n"
     ]
    },
    {
     "ename": "",
     "evalue": "",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31mThe Kernel crashed while executing code in the the current cell or a previous cell. Please review the code in the cell(s) to identify a possible cause of the failure. Click <a href='https://aka.ms/vscodeJupyterKernelCrash'>here</a> for more info. View Jupyter <a href='command:jupyter.viewOutput'>log</a> for further details."
     ]
    }
   ],
   "source": [
    "from neuromatch.variables.regneuron import RegisteredNeuron\n",
    "from neuromatch.variables import AllToAllList\n",
    "import pickle\n",
    "with open(r\"E:\\Anaconda\\envs\\maze\\Lib\\site-packages\\neuromatch\\temp.pkl\", 'rb') as f:\n",
    "    index_map, ref_indexmaps, ata_p_sames, ata_indexmaps = pickle.load(f)\n",
    "\n",
    "reg_neuron = RegisteredNeuron(index_line=index_map[:, 2], ref_indexmaps=ref_indexmaps, ata_p_sames=AllToAllList(ata_p_sames), ata_indexmaps=AllToAllList(ata_indexmaps))\n",
    "print(reg_neuron.ori_content)\n",
    "reg_neuron.optimize()\n",
    "print(reg_neuron.opt_content)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "maze11",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.11"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}