Skip to content

Commit c07c45e

Browse files
committed
WIP: Move from q_vec -> s; s -> q in the code.
The opt_broadcast.py still needs to be updated; issues with running Karimi baseline.
1 parent 87c47e1 commit c07c45e

File tree

5 files changed

+335
-344
lines changed

5 files changed

+335
-344
lines changed

RedQueen-demo.ipynb

Lines changed: 33 additions & 63 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,12 @@
33
{
44
"cell_type": "code",
55
"execution_count": 1,
6-
"metadata": {
7-
"collapsed": false
8-
},
6+
"metadata": {},
97
"outputs": [
108
{
119
"data": {
1210
"text/plain": [
13-
"<matplotlib.figure.Figure at 0x7f5d6e0ca390>"
11+
"<matplotlib.figure.Figure at 0x7f68dee2a160>"
1412
]
1513
},
1614
"metadata": {},
@@ -31,10 +29,8 @@
3129
},
3230
{
3331
"cell_type": "code",
34-
"execution_count": 3,
35-
"metadata": {
36-
"collapsed": false
37-
},
32+
"execution_count": 2,
33+
"metadata": {},
3834
"outputs": [],
3935
"source": [
4036
"import redqueen.utils as U\n",
@@ -101,17 +97,15 @@
10197
},
10298
{
10399
"cell_type": "code",
104-
"execution_count": 4,
105-
"metadata": {
106-
"collapsed": false
107-
},
100+
"execution_count": 3,
101+
"metadata": {},
108102
"outputs": [],
109103
"source": [
110104
"sim_opts_1_follower = OM.SimOpts(\n",
111105
" src_id=0,\n",
112106
" end_time=100,\n",
113-
" q_vec=np.array([1]),\n",
114-
" s=1.0,\n",
107+
" s=np.array([1]),\n",
108+
" q=1.0,\n",
115109
" other_sources=[('Hawkes', {'src_id': 1, 'seed': 1, 'l_0': 1.0, 'alpha': 1.0, 'beta': 5.0})],\n",
116110
" sink_ids=[1000],\n",
117111
" edge_list=[(0, 1000), (1, 1000)]\n",
@@ -129,17 +123,15 @@
129123
},
130124
{
131125
"cell_type": "code",
132-
"execution_count": 5,
133-
"metadata": {
134-
"collapsed": false
135-
},
126+
"execution_count": 4,
127+
"metadata": {},
136128
"outputs": [
137129
{
138130
"name": "stdout",
139131
"output_type": "stream",
140132
"text": [
141-
"CPU times: user 83.3 ms, sys: 3.33 ms, total: 86.7 ms\n",
142-
"Wall time: 84.6 ms\n"
133+
"CPU times: user 66.7 ms, sys: 0 ns, total: 66.7 ms\n",
134+
"Wall time: 63.1 ms\n"
143135
]
144136
}
145137
],
@@ -154,24 +146,22 @@
154146
" 'type': 'Opt',\n",
155147
" 'seed': seed,\n",
156148
" 'capacity': num_opt_tweets,\n",
157-
" 's': sim_opts_1_follower.s\n",
149+
" 'q': sim_opts_1_follower.q\n",
158150
"}\n",
159151
"OR.add_perf(perf_opt, opt_df, sim_opts_1_follower)"
160152
]
161153
},
162154
{
163155
"cell_type": "code",
164-
"execution_count": 6,
165-
"metadata": {
166-
"collapsed": false
167-
},
156+
"execution_count": 5,
157+
"metadata": {},
168158
"outputs": [
169159
{
170160
"name": "stdout",
171161
"output_type": "stream",
172162
"text": [
173-
"CPU times: user 73.3 ms, sys: 0 ns, total: 73.3 ms\n",
174-
"Wall time: 72 ms\n"
163+
"CPU times: user 100 ms, sys: 6.67 ms, total: 107 ms\n",
164+
"Wall time: 103 ms\n"
175165
]
176166
}
177167
],
@@ -186,7 +176,7 @@
186176
" 'type': 'Poisson',\n",
187177
" 'seed': seed,\n",
188178
" 'capacity': num_poisson_tweets,\n",
189-
" 's': sim_opts_1_follower.s\n",
179+
" 'q': sim_opts_1_follower.q\n",
190180
"}\n",
191181
"OR.add_perf(perf_poisson, poisson_df, sim_opts_1_follower)"
192182
]
@@ -203,9 +193,7 @@
203193
{
204194
"cell_type": "code",
205195
"execution_count": 7,
206-
"metadata": {
207-
"collapsed": false
208-
},
196+
"metadata": {},
209197
"outputs": [
210198
{
211199
"name": "stdout",
@@ -223,9 +211,7 @@
223211
{
224212
"cell_type": "code",
225213
"execution_count": 8,
226-
"metadata": {
227-
"collapsed": false
228-
},
214+
"metadata": {},
229215
"outputs": [
230216
{
231217
"name": "stdout",
@@ -244,9 +230,7 @@
244230
{
245231
"cell_type": "code",
246232
"execution_count": 9,
247-
"metadata": {
248-
"collapsed": false
249-
},
233+
"metadata": {},
250234
"outputs": [
251235
{
252236
"data": {
@@ -274,9 +258,7 @@
274258
{
275259
"cell_type": "code",
276260
"execution_count": 10,
277-
"metadata": {
278-
"collapsed": false
279-
},
261+
"metadata": {},
280262
"outputs": [
281263
{
282264
"name": "stdout",
@@ -324,9 +306,7 @@
324306
{
325307
"cell_type": "code",
326308
"execution_count": 11,
327-
"metadata": {
328-
"collapsed": false
329-
},
309+
"metadata": {},
330310
"outputs": [
331311
{
332312
"data": {
@@ -354,9 +334,7 @@
354334
{
355335
"cell_type": "code",
356336
"execution_count": 12,
357-
"metadata": {
358-
"collapsed": false
359-
},
337+
"metadata": {},
360338
"outputs": [
361339
{
362340
"data": {
@@ -607,10 +585,8 @@
607585
},
608586
{
609587
"cell_type": "code",
610-
"execution_count": 13,
611-
"metadata": {
612-
"collapsed": false
613-
},
588+
"execution_count": 11,
589+
"metadata": {},
614590
"outputs": [],
615591
"source": [
616592
"@Deco.optioned('opts')\n",
@@ -646,10 +622,8 @@
646622
},
647623
{
648624
"cell_type": "code",
649-
"execution_count": 14,
650-
"metadata": {
651-
"collapsed": false
652-
},
625+
"execution_count": 12,
626+
"metadata": {},
653627
"outputs": [],
654628
"source": [
655629
"example_1 = perf_to_json([opt_df, poisson_df], ['redqueen', 'poisson'], \n",
@@ -658,18 +632,16 @@
658632
},
659633
{
660634
"cell_type": "code",
661-
"execution_count": 15,
662-
"metadata": {
663-
"collapsed": false
664-
},
635+
"execution_count": 13,
636+
"metadata": {},
665637
"outputs": [
666638
{
667639
"data": {
668640
"text/plain": [
669641
"43.482480948594869"
670642
]
671643
},
672-
"execution_count": 15,
644+
"execution_count": 13,
673645
"metadata": {},
674646
"output_type": "execute_result"
675647
}
@@ -681,9 +653,7 @@
681653
{
682654
"cell_type": "code",
683655
"execution_count": 16,
684-
"metadata": {
685-
"collapsed": false
686-
},
656+
"metadata": {},
687657
"outputs": [
688658
{
689659
"data": {
@@ -1894,7 +1864,7 @@
18941864
"name": "python",
18951865
"nbconvert_exporter": "python",
18961866
"pygments_lexer": "ipython3",
1897-
"version": "3.5.2"
1867+
"version": "3.6.1"
18981868
}
18991869
},
19001870
"nbformat": 4,

0 commit comments

Comments
 (0)