#include <util/generic/string.h>

namespace Datacloud {
namespace MLUtils {
namespace KMeans {

i64 Expectation(
    const TString& ytToken,
    const TString& cluster,
    i64 nClusters,
    const TString& clusterCetnersTable,
    const TString& objectsTable,
    const TString& outputExpectationTable);

void Minimization(
    const TString& ytToken,
    const TString& cluster,
    const TString& expectationTable,
    const TString& outputMinimizationTable);

void ExpectationMinimization(
    const TString& ytToken,
    const TString& cluster,
    i64 nClusters,
    const TString& objectsTable,
    const TString& clusterCentersTable,
    const TString& outputClusterCentersTable);

}  // namespace KMeans
}  // namespace MLUtils
}  // namesapce Datacloud
