![]() |
UFJF - Machine Learning Toolkit
0.51.8
|
Wrapper for the implementation of the K-Means clustering algorithm. More...
#include <KMeans.hpp>
Public Member Functions | |
KMeans (const Data< T > &samples, size_t k, std::string initialization="random", size_t seed=0, int verbose=0) | |
bool | train () override |
Function that execute the training phase of a Learner. More... | |
double | assign_clusters (const std::vector< mltk::PointPointer< T >> &points) |
double | clusters_variance (const Clusters &clusters) |
void | compute_centers () |
double | evaluate (const Point< T > &p, bool raw_value=false) override |
Returns the class of a feature point based on the trained Learner. More... | |
std::vector< mltk::Point< size_t > > | clusters () override |
std::string | getFormulationString () override |
getFormulationString Returns a string that represents the formulation of the learner (Primal or Dual). More... | |
![]() | |
Clusterer (DataPointer< double > samples=nullptr, size_t clusters=0) | |
std::vector< mltk::Point< double > > | centers () |
![]() | |
Learner (DataPointer< T > _samples) | |
Learner (const Learner< T > &learner) | |
virtual mltk::Point< double > | batchEvaluate (const Data< T > &data) |
evaluate a batch of points. More... | |
auto | getSamples () |
Get the Data used by the learner. More... | |
double | getElapsedTime () const |
Get the elapsed time in the training phase of the Learner. More... | |
int | getCtot () const |
Get the total number of updates of the Learner. More... | |
int | getSteps () const |
getSteps Returns the number of steps through the data by the Learner. More... | |
int | getUpdates () const |
getUpdates Returns the number of updates needed to get to the the solution. More... | |
double | getMaxTime () const |
getMaxTime Returns the maximum running time in the training phase of the Learner. More... | |
double | getPredictionProbability () const |
Get the probability of the last prediction. More... | |
void | setSeed (const size_t _seed) |
Set the seed to be used by the learner. More... | |
virtual void | setSamples (const Data< T > &data) |
setSamples Set the samples used by the Learner. More... | |
virtual void | setSamples (DataPointer< T > data) |
setSamples Set the samples used by the Learner. More... | |
void | setTimer (Timer _timer) |
setTimer Set the timer used by the Learner. More... | |
void | setSteps (int _steps) |
Set the partial number of steps used in the training phase of the Learner. More... | |
void | setCtot (int _ctot) |
Set the partial number of updates of the Learner. More... | |
void | setVerbose (int _verbose) |
Set the level of verbose. More... | |
void | setStartTime (double stime) |
setStartTime Set the initial time of the Learner. More... | |
void | setMaxTime (double maxtime) |
Set the max time of execution. More... | |
void | setEPS (double eps) |
setEPS Set the precision of the Learner. More... | |
void | setMaxIterations (int max_it) |
setMaxIterations Set the max number of iterations of the Learner. More... | |
void | setMaxEpochs (int MAX_EPOCHS) |
Set the max number of epochs for the learner training. More... | |
void | setMaxUpdates (int max_up) |
setMaxIterations Set the max number of updates of the Learner. More... | |
void | setLearningRate (double learning_rate) |
Set the learning rate of the Learner. More... | |
Additional Inherited Members | |
![]() | |
metrics::dist::Euclidean< double > | dist_function |
Function used to compute the metrics between two points. More... | |
size_t | n_clusters |
Number of clusters for the cluster method. More... | |
std::vector< mltk::Point< double > > | m_centers |
Vector with the centers of the clusters. More... | |
std::vector< mltk::Point< size_t > > | m_clusters |
Clusters of points. More... | |
![]() | |
std::shared_ptr< Data< T > > | samples |
Samples used in the model training. More... | |
double | rate = 0.5f |
Learning rate. More... | |
double | start_time = 0.0f |
Initial time. More... | |
double | max_time = 110 |
Maximum time of training. More... | |
int | steps = 0 |
Number of steps in the data. More... | |
int | ctot = 0 |
Number of updates of the weights. More... | |
double | EPS = 0.0000001 |
Max precision. More... | |
double | MIN_INC = 1.001 |
Minimun Increment. More... | |
int | MAX_IT = 1E9 |
Max number of iterations. More... | |
int | MAX_UP = 1E9 |
Max number of updates. More... | |
int | MAX_EPOCH = 1E9 |
int | verbose = 1 |
Verbose level of the output. More... | |
Timer | timer = Timer() |
Timer used to measure the time elapsed in the execution of a Learner. More... | |
size_t | seed = 0 |
seed for random operations. More... | |
double | pred_prob = 1.0 |
Wrapper for the implementation of the K-Means clustering algorithm.
|
overridevirtual |
Returns the class of a feature point based on the trained Learner.
p | Point to be evaluated. |
Implements mltk::Learner< T >.
|
overridevirtual |
getFormulationString Returns a string that represents the formulation of the learner (Primal or Dual).
Implements mltk::Learner< T >.
|
overridevirtual |