Options.hpp

Go to the documentation of this file.

#ifndef UMAPPP_OPTIONS_HPP
#define UMAPPP_OPTIONS_HPP
 
#include <cstdint>
 
namespace umappp {
 
enum InitializeMethod : char { SPECTRAL, SPECTRAL_ONLY, RANDOM, NONE };
 
struct Options {
    double local_connectivity = 1;
 
    double bandwidth = 1;
 
    double mix_ratio = 1;
 
    double spread = 1;
 
    double min_dist = 0.1;
 
    double a = 0;
 
    double b = 0;
 
    double repulsion_strength = 1;
 
    InitializeMethod initialize = InitializeMethod::SPECTRAL;
 
    int num_epochs = -1;
 
    double learning_rate = 1; 
 
    double negative_sample_rate = 5;
 
    int num_neighbors = 15;
 
    uint64_t seed = 1234567890;
 
    int num_threads = 1;
 
    int parallel_optimization = false;
};
 
}
 
#endif