Options.hpp

Go to the documentation of this file.

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