#include "pycluster.h"


void init_clustering(py::module &m) {

    m.def("hcluster", [](SparseMatrixWrapper &smw, py::object centers, double beta,
                         py::object msr, py::object weights, double eps,
                         uint64_t kmeansmaxiter, uint64_t seed, py::ssize_t mbsize, py::ssize_t ncheckins,
                         py::ssize_t reseed_count, bool with_rep, bool use_cs) {
                             return __py_cluster_from_centers(smw, centers, beta, msr, weights, eps, kmeansmaxiter,
                                 seed,
                                 mbsize, ncheckins, reseed_count, with_rep, use_cs);
                         },
    py::arg("smw"),
    py::arg("centers"),
    py::arg("prior") = 0.,
    py::arg("msr") = 2,
    py::arg("weights") = py::none(),
    py::arg("eps") = 1e-10,
    py::arg("maxiter") = 100,
    py::arg("seed") = 0,
    py::arg("mbsize") = py::ssize_t(-1),
    py::arg("ncheckins") = py::ssize_t(-1),
    py::arg("reseed_count") = py::ssize_t(5),
    py::arg("with_rep") = false, py::arg("cs") = false,
    "Clusters a SparseMatrixWrapper object using settings and the centers provided above; set prior to < 0 for it to be 1 / ncolumns(). Performs seeding, followed by EM or minibatch k-means");
} // init_clustering