@Override
  public void prepare(GraphDatabaseService database, final Map<String, Object> params) {
    generatorConfiguration =
        new BasicGeneratorConfig(
            new BarabasiAlbertRelationshipGenerator(
                new BarabasiAlbertConfig((Integer) params.get(NETWORK_SIZE), 3)),
            SocialNetworkNodeCreator.getInstance(),
            SocialNetworkRelationshipCreator.getInstance()) {

          @Override
          public int getBatchSize() {
            return (int) params.get(BATCH_SIZE);
          }
        };
  }
 private GeneratorConfiguration getGeneratorConfiguration(int numberOfNodes, int numberOfEdges) {
   return new BasicGeneratorConfig(
       new ErdosRenyiRelationshipGenerator(new ErdosRenyiConfig(numberOfNodes, numberOfEdges)),
       SocialNetworkNodeCreator.getInstance(),
       SocialNetworkRelationshipCreator.getInstance());
 }