33#==========================================================================
44def neighborhood (graph , src , num_hops ):
55 num_nodes = graph .nrows
6- w = grb .Vector .sparse (grb .types . BOOL , num_nodes )
7- v = grb .Vector .sparse (grb .types . BOOL , num_nodes )
6+ w = grb .Vector .sparse (grb .BOOL , num_nodes )
7+ v = grb .Vector .sparse (grb .BOOL , num_nodes )
88 w [src ] = True
99 v .assign_scalar (True , mask = w )
1010
@@ -19,12 +19,12 @@ def neighborhood(graph, src, num_hops):
1919def pagerank (A , damping = 0.85 , tol = 1e-4 , itermax = 100 ):
2020 n = A .nrows
2121
22- r = grb .Vector .dense (grb .types . FP32 , n , fill = 1.0 / n )
23- t = grb .Vector .dense (grb .types . FP32 , n )
24- d = grb .Vector .dense (grb .types . FP32 , n , fill = damping )
22+ r = grb .Vector .dense (grb .FP32 , n , fill = 1.0 / n )
23+ t = grb .Vector .dense (grb .FP32 , n )
24+ d = grb .Vector .dense (grb .FP32 , n , fill = damping )
2525
26- A .reduce_vector (out = d , accum = grb .types . FP32 .DIV ,
27- mon = grb .types . FP32 .PLUS_MONOID )
26+ A .reduce_vector (out = d , accum = grb .FP32 .DIV ,
27+ mon = grb .FP32 .PLUS_MONOID )
2828
2929 teleport = (1 - damping ) / n
3030
@@ -34,12 +34,12 @@ def pagerank(A, damping = 0.85, tol = 1e-4, itermax = 100):
3434 r [:] = teleport
3535 A .mxv (w ,
3636 out = r ,
37- accum = grb .types . FP32 .PLUS ,
38- semiring = grb .types . FP32 .PLUS_SECOND ,
37+ accum = grb .FP32 .PLUS ,
38+ semiring = grb .FP32 .PLUS_SECOND ,
3939 desc = grb .descriptor .T0 )
4040 t -= r
41- t .apply (grb .types . FP32 .ABS , out = t )
42- rdiff = t .reduce_float (grb .types . FP32 .PLUS_MONOID )
41+ t .apply (grb .FP32 .ABS , out = t )
42+ rdiff = t .reduce_float (grb .FP32 .PLUS_MONOID )
4343 if rdiff <= tol :
4444 break
4545 return r
@@ -55,9 +55,9 @@ def triangle_count(A):
5555 A_nl = A_unw .offdiag ()
5656
5757 # make sure of symmetry
58- A_nl .eadd (A_nl , add_op = grb .types . UINT64 .LOR , desc = grb .descriptor .T1 , out = A_nl )
58+ A_nl .eadd (A_nl , add_op = grb .UINT64 .LOR , desc = grb .descriptor .T1 , out = A_nl )
5959
6060 # count triangles
61- C = A_nl .mxm (A_nl , semiring = grb .types . UINT64 .PLUS_TIMES , mask = A_nl )
62- count = C .reduce_int (grb .types . UINT64 .PLUS_MONOID )
61+ C = A_nl .mxm (A_nl , semiring = grb .UINT64 .PLUS_TIMES , mask = A_nl )
62+ count = C .reduce_int (grb .UINT64 .PLUS_MONOID )
6363 return (int )(count / 6.0 )
0 commit comments