forked from zheng-yp/DecoupledDGNN
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpropagation.pyx
More file actions
executable file
·19 lines (13 loc) · 892 Bytes
/
propagation.pyx
File metadata and controls
executable file
·19 lines (13 loc) · 892 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from propagation cimport Instantgnn
cdef class InstantGNN:
cdef Instantgnn c_instantgnn
def __cinit__(self):
self.c_instantgnn=Instantgnn()
def initial_operation(self,path,dataset,unsigned int m,unsigned int n,rmax,alpha,np.ndarray array3):
return self.c_instantgnn.initial_operation(path.encode(),dataset.encode(),m,n,rmax,alpha,Map[MatrixXd](array3))
def dynamic_operation(self, unsigned int v_from,unsigned int v_to, rmax,alpha, np.ndarray array3):
return self.c_instantgnn.dynamic_operation(v_from, v_to, rmax, alpha, Map[MatrixXd](array3))
def snapshot_operation(self, upfile, rmax,alpha, np.ndarray array3):
return self.c_instantgnn.snapshot_operation(upfile.encode(), rmax, alpha, Map[MatrixXd](array3))
def overall_operation(self, rmax,alpha, np.ndarray array3):
return self.c_instantgnn.overall_operation(rmax, alpha, Map[MatrixXd](array3))