#include "daal.h"
#include "service.h"
using namespace std;
using namespace daal;
using namespace daal::algorithms;
typedef float algorithmFPType;
string datasetFileName = "../data/batch/kmeans_csr.csv";
const size_t nClusters = 20;
int main(int argc, char *argv[])
{
checkArguments(argc, argv, 1, &datasetFileName);
CSRNumericTablePtr dataTable(createSparseTable<float>(datasetFileName));
kmeans::init::Batch<algorithmFPType, kmeans::init::randomCSR> init(nClusters);
init.input.set(kmeans::init::data, dataTable);
init.compute();
NumericTablePtr centroids = init.getResult()->get(kmeans::init::centroids);
kmeans::Batch<algorithmFPType, kmeans::lloydCSR> algorithm(nClusters, 0);
algorithm.input.set(kmeans::data, dataTable);
algorithm.input.set(kmeans::inputCentroids, centroids);
algorithm.compute();
printNumericTable(algorithm.getResult()->get(kmeans::assignments), "First 10 cluster assignments:", 10);
return 0;
}