In [1]:

    

import random as r
import numpy as np
import sys
sys.path.append('../src/mane/')


    

In [7]:

    

import graph as g
reload(g)
fb = g.graph_from_pickle('../src/mane/data/egonets.graph')


    

In [8]:

    

gen = fb.gen_contrast2()


    

In [9]:

    

test = next(gen)


    

    




WARNING: Short negative samples. 13

In [10]:

    

test


    

    
Out[10]:





({'class': array([2955, 2875, 3385, ..., 1219, 1017, 3011], dtype=int32),
  'target': array([3043, 3043, 3043, ...,  996,  996,  996], dtype=int32)},
 {'label': array([ 1.,  1.,  0., ...,  0.,  0.,  0.], dtype=float32)})

In [11]:

    

len(test[0]['class'])


    

    
Out[11]:





1700