mirror of
https://github.com/Relintai/pmlpp.git
synced 2025-01-10 17:49:36 +01:00
48 lines
973 B
C++
48 lines
973 B
C++
|
|
#ifndef MLPP_K_MEANS_H
|
|
#define MLPP_K_MEANS_H
|
|
|
|
//
|
|
// KMeans.hpp
|
|
//
|
|
// Created by Marc Melikyan on 10/2/20.
|
|
//
|
|
|
|
#include "core/math/math_defs.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
|
|
class MLPPKMeans {
|
|
public:
|
|
MLPPKMeans(std::vector<std::vector<real_t>> inputSet, int k, std::string init_type = "Default");
|
|
std::vector<std::vector<real_t>> modelSetTest(std::vector<std::vector<real_t>> X);
|
|
std::vector<real_t> modelTest(std::vector<real_t> x);
|
|
void train(int epoch_num, bool UI = 1);
|
|
real_t score();
|
|
std::vector<real_t> silhouette_scores();
|
|
|
|
private:
|
|
void Evaluate();
|
|
void computeMu();
|
|
|
|
void centroidInitialization(int k);
|
|
void kmeansppInitialization(int k);
|
|
real_t Cost();
|
|
|
|
std::vector<std::vector<real_t>> inputSet;
|
|
std::vector<std::vector<real_t>> mu;
|
|
std::vector<std::vector<real_t>> r;
|
|
|
|
real_t euclideanDistance(std::vector<real_t> A, std::vector<real_t> B);
|
|
|
|
real_t accuracy_threshold;
|
|
int k;
|
|
|
|
std::string init_type;
|
|
};
|
|
|
|
|
|
#endif /* KMeans_hpp */
|