Program Listing for File StochasticBlockmodel.hpp¶

/*
 * StochasticBlockmodel.hpp
 *
 *  Created on: 13.08.2014
 *      Author: Christian Staudt
 */

#ifndef NETWORKIT_GENERATORS_STOCHASTIC_BLOCKMODEL_HPP_
#define NETWORKIT_GENERATORS_STOCHASTIC_BLOCKMODEL_HPP_

#include <networkit/generators/StaticGraphGenerator.hpp>

namespace NetworKit {

class StochasticBlockmodel final : public StaticGraphGenerator {

public:
    StochasticBlockmodel(count n, count nBlocks, const std::vector<index> &membership,
                         const std::vector<std::vector<double>> &affinity);

    Graph generate() override;

private:
    count n;
    std::vector<index> membership;
    const std::vector<std::vector<double>> &affinity;
};

} /* namespace NetworKit */
#endif // NETWORKIT_GENERATORS_STOCHASTIC_BLOCKMODEL_HPP_