@@ -111,11 +111,21 @@ def add_SIS(self, infected, beta, lbd):
111
111
data = {'infected' : infected , 'beta' : beta , 'lambda' : lbd , 'token' : self .token })
112
112
return res .status_code
113
113
114
+ def add_SEIS (self , infected , beta , alpha , lbd ):
115
+ res = put ('%s/api/SEIS' % self .base ,
116
+ data = {'infected' : infected , 'beta' : beta , 'lambda' : lbd , 'alpha' : alpha , 'token' : self .token })
117
+ return res .status_code
118
+
114
119
def add_SIR (self , infected , beta , gamma ):
115
120
res = put ('%s/api/SIR' % self .base ,
116
121
data = {'infected' : infected , 'beta' : beta , 'gamma' : gamma , 'token' : self .token })
117
122
return res .status_code
118
123
124
+ def add_SEIR (self , infected , beta , gamma , alpha ):
125
+ res = put ('%s/api/SEIR' % self .base ,
126
+ data = {'infected' : infected , 'beta' : beta , 'gamma' : gamma , 'alpha' : alpha , 'token' : self .token })
127
+ return res .status_code
128
+
119
129
def add_independent_cascades (self , infected ):
120
130
res = put ('%s/api/IndependentCascades' % self .base , data = {'infected' : infected , 'token' : self .token })
121
131
return res .status_code
@@ -212,7 +222,7 @@ def load_exploratory(self, name):
212
222
res = post ('%s/api/Exploratory' % self .base , data = {'exploratory' : name , 'token' : self .token })
213
223
return res .status_code
214
224
215
- def upload_graph (self , directed , graph ):
225
+ def upload_graph (self , directed , graph , dynamic = False ):
216
226
res = post ('%s/api/UploadNetwork' % self .base ,
217
- data = {'directed' : directed , 'graph' : json .dumps (graph ), 'token' : self .token })
227
+ data = {'directed' : directed , 'graph' : json .dumps (graph ), 'dynamic' : dynamic , ' token' : self .token })
218
228
return res .status_code
0 commit comments