Skip to content

Commit 2204417

Browse files
rikabrikab
committed
unbalanced_ot
1 parent c12cf84 commit 2204417

File tree

9 files changed

+26545
-48
lines changed

9 files changed

+26545
-48
lines changed

‎examples/pairwise_semds_examples.ipynb‎

Lines changed: 4964 additions & 31 deletions
Large diffs are not rendered by default.

‎examples/specter_vs_shaper.ipynb‎

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,18 @@
22
"cells": [
33
{
44
"cell_type": "code",
5-
"execution_count": 11,
5+
"execution_count": 2,
66
"metadata": {},
77
"outputs": [
88
{
9-
"name": "stdout",
10-
"output_type": "stream",
11-
"text": [
12-
"Using CPU!\n"
9+
"ename": "ImportError",
10+
"evalue": "cannot import name 'load_cmsopendata' from 'pyspecter.utils.data_utils' (/home/rikab/miniconda3/envs/SHAPER/lib/python3.8/site-packages/pyspecter/utils/data_utils.py)",
11+
"output_type": "error",
12+
"traceback": [
13+
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
14+
"\u001b[0;31mImportError\u001b[0m Traceback (most recent call last)",
15+
"Cell \u001b[0;32mIn[2], line 5\u001b[0m\n\u001b[1;32m 2\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mpyshaper\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mShaper\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m Shaper\n\u001b[1;32m 4\u001b[0m \u001b[38;5;66;03m# Utils\u001b[39;00m\n\u001b[0;32m----> 5\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mpyspecter\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mutils\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mdata_utils\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m load_cmsopendata, load_triangles, random_triangles, kT_N\n\u001b[1;32m 6\u001b[0m \u001b[38;5;28;01mfrom\u001b[39;00m \u001b[38;5;21;01mpyspecter\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mutils\u001b[39;00m\u001b[38;5;21;01m.\u001b[39;00m\u001b[38;5;21;01mplot_utils\u001b[39;00m \u001b[38;5;28;01mimport\u001b[39;00m initialize, newplot, plot_event\n\u001b[1;32m 7\u001b[0m initialize()\n",
16+
"\u001b[0;31mImportError\u001b[0m: cannot import name 'load_cmsopendata' from 'pyspecter.utils.data_utils' (/home/rikab/miniconda3/envs/SHAPER/lib/python3.8/site-packages/pyspecter/utils/data_utils.py)"
1317
]
1418
}
1519
],
@@ -41,7 +45,7 @@
4145
},
4246
{
4347
"cell_type": "code",
44-
"execution_count": 12,
48+
"execution_count": null,
4549
"metadata": {},
4650
"outputs": [
4751
{
@@ -73,7 +77,7 @@
7377
"pad = 100 # Note that runtime is pad^2, memory is pad^4\n",
7478
"\n",
7579
"dataset = \"CMS\"\n",
76-
"n_samples = 2500*2\n",
80+
"n_samples = 15*2\n",
7781
"batch_size = 1 # Number of pairs to process in parallel, can probably be increased.\n",
7882
"\n",
7983
"if dataset == \"CMS\":\n",

‎examples/test_event.csv‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
1.946833582098391434e-03,-2.506669094896074590e+00,6.858044595927288434e-01
2+
2.712308668951239274e-02,-2.395733324316973167e+00,-7.051335131660485533e-01
3+
4.252628522884490531e-03,-1.593388763471011549e+00,-1.663124583460474648e-01
4+
8.671102361505395040e-03,-7.863425166247077769e-01,-8.710517379837194873e-01
5+
2.555823425571315810e-02,-6.475302435524588063e-01,-3.225731560122301889e-01
6+
1.842934053626422986e-02,-8.287331012669834074e-01,-5.206061923551966730e-01
7+
2.407543462789804925e-02,-9.068720937976867402e-01,-2.476151180331047086e-01
8+
2.330772331302695833e-03,-2.276813503659568916e+00,-1.123376198429042905e+00
9+
1.734830238127772761e-02,-2.418970912432052867e+00,-1.557294656215324702e+00
10+
3.669693786717157045e-03,1.138829993223218429e+00,-1.542752636488338602e+00
11+
3.568804801689226362e-03,1.146502706187725540e+00,1.405105375478561160e-02
12+
2.481030427331071958e-03,-3.673700288991282203e+00,-1.117680114594971519e+00
13+
6.341361760082647216e-03,-3.258763486160965783e+00,-4.303901841694475205e-01
14+
3.151507824584277601e-03,-2.457911244315481358e+00,-1.142311905103259306e+00
15+
4.949140194672593460e-03,-1.299816246840287182e+00,3.849545010237198639e-01
16+
2.012207364980487471e-03,-3.533860956928595254e+00,-2.498129457208310422e-01
17+
2.896164333661268436e-02,-1.841976383800170813e+00,1.075392218413303014e-01
18+
9.892048315007125059e-02,-1.739076623048157799e+00,9.589347818019285097e-02
19+
3.209326297406507600e-02,-1.492572693185503896e+00,1.518312725073694136e-01
20+
1.731865792267392110e-02,-1.662961526486048669e+00,6.547447957936588514e-01
21+
2.724059941064443588e-02,-6.693472831274321910e-01,2.331949499077416721e-01
22+
8.378476522706332363e-03,-4.759501692946215279e-01,8.407167718497893238e-01
23+
4.660803810872238100e-03,-2.338504500401682451e-01,-4.187627584738236319e-02
24+
4.898639547408260853e-03,-1.164966947640825090e+00,-1.258278504141162157e+00
25+
1.992905340627575735e-03,-2.682120175259854467e+00,-1.522841709125863918e+00
26+
4.858508345436065266e-03,-2.634072389540283776e+00,3.485659606879418781e-01
27+
2.109344718202972938e-02,-4.592842613558195120e-01,8.961051769279103352e-02
28+
8.811717570884416983e-02,-5.241475289628073631e-01,3.330371333293042357e-01
29+
1.642666920804896069e-01,-5.021837277120528853e-01,3.233565010277199003e-01
30+
5.293469055326138761e-02,-5.495390347965926070e-01,4.536757078680420663e-01
31+
1.429412952976664118e-02,-6.149292973667349615e-01,-1.126211776867826891e-01
32+
1.313773863776025209e-02,-3.852181233984447450e-01,-3.521588202972340542e-01
33+
1.906543172967205572e-03,-9.847308191535980582e-02,5.560913411480620239e-01
34+
6.980718812209397994e-02,4.018839500006894383e-01,-1.159558115477002105e-01
35+
6.113821484848732529e-02,-2.940122648579881037e+00,-5.836900102410805952e-02
36+
4.077522521663994731e-02,-2.800110033841963109e+00,1.661176064526000140e-01
37+
1.576491102967581904e-03,-3.171044333535768533e+00,1.277169474512477976e+00
38+
5.743237327166038179e-03,-1.424547405596528593e+00,-9.149044388905999137e-01
39+
3.122126657277324302e-02,-1.604535073177100424e+00,-3.267621311061951195e-01
40+
3.438967633846304241e-03,-3.012951599867320862e+00,-3.908149183633719548e-01
41+
2.195992624969552148e-03,-3.779148563358314838e+00,3.933985273053383080e-01
42+
1.469147053813531743e-02,1.131798211912675045e+00,-4.426586758097920127e-01
43+
1.246182776157616842e-02,1.653598021175882504e+00,-2.898354631970346262e-01
44+
1.615597640885492112e-03,2.364056845519633576e+00,1.232500962541132283e+00
45+
3.778206866526858613e-03,4.216110559677100356e-01,1.036080143964499323e+00
46+
1.845018876219109557e-03,-2.930306054903271396e-02,-7.890840520671498215e-01
47+
8.727416192962245134e-03,-2.346746658700403776e-01,1.529241100923383945e+00
48+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
49+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
50+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
51+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
52+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
53+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
54+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
55+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
56+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
57+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
58+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
59+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
60+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
61+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
62+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
63+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
64+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
65+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
66+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
67+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
68+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
69+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
70+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
71+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
72+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
73+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
74+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
75+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
76+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
77+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
78+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
79+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
80+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
81+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
82+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
83+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
84+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
85+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
86+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
87+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
88+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
89+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
90+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
91+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
92+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
93+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
94+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00
95+
0.000000000000000000e+00,0.000000000000000000e+00,0.000000000000000000e+00

0 commit comments

Comments
 (0)