Options.hpp

Go to the documentation of this file.

#ifndef MNNCORRECT_OPTIONS_HPP
#define MNNCORRECT_OPTIONS_HPP
 
#include "knncolle/knncolle.hpp"
#include "ReferencePolicy.hpp"
 
namespace mnncorrect {
 
template<typename Dim_ = int, typename Index_ = int, typename Float_ = double>
struct Options {
    int num_neighbors = 15;
 
    double num_mads = 3;
 
    std::shared_ptr<knncolle::Builder<knncolle::SimpleMatrix<Dim_, Index_, Float_>, Float_> > builder;
 
    std::vector<size_t> order;
 
    bool automatic_order = true;
 
    int robust_iterations = 2;
 
    double robust_trim = 0.25;
 
    ReferencePolicy reference_policy = ReferencePolicy::MAX_RSS;
 
    size_t mass_cap = -1;
 
    int num_threads = 1;
};
 
}
 
#endif