Stochastic Block Model Python