data_set_variants.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_DETAIL_CMD_DATA_SET_VARIANTS_HPP_
#define PLSSVM_DETAIL_CMD_DATA_SET_VARIANTS_HPP_
#pragma once
 
#include "plssvm/data_set.hpp"                   // plssvm::data_set
#include "plssvm/detail/cmd/parser_predict.hpp"  // plssvm::detail::cmd::parser_predict
#include "plssvm/detail/cmd/parser_scale.hpp"    // plssvm::detail::cmd::parser_scale
#include "plssvm/detail/cmd/parser_train.hpp"    // plssvm::detail::cmd::parser_train
 
#include <string>                                // std::string
#include <variant>                               // std::variant
 
namespace plssvm::detail::cmd {
 
// clang-format off
using data_set_variants = std::variant<plssvm::data_set<float, int>, plssvm::data_set<float, std::string>,
                                       plssvm::data_set<double, int>, plssvm::data_set<double, std::string>>;
// clang-format on
 
template <typename real_type, typename label_type = typename data_set<real_type>::label_type>
[[nodiscard]] inline data_set_variants data_set_factory_impl(const cmd::parser_train &cmd_parser) {
    return data_set_variants{ plssvm::data_set<real_type, label_type>{ cmd_parser.input_filename } };
}
template <typename real_type, typename label_type = typename data_set<real_type>::label_type>
[[nodiscard]] inline data_set_variants data_set_factory_impl(const cmd::parser_predict &cmd_parser) {
    return data_set_variants{ plssvm::data_set<real_type, label_type>{ cmd_parser.input_filename } };
}
template <typename real_type, typename label_type = typename data_set<real_type>::label_type>
[[nodiscard]] inline data_set_variants data_set_factory_impl(const cmd::parser_scale &cmd_parser) {
    if (!cmd_parser.restore_filename.empty()) {
        return data_set_variants{ plssvm::data_set<real_type, label_type>{ cmd_parser.input_filename, { cmd_parser.restore_filename } } };
    } else {
        return data_set_variants{ plssvm::data_set<real_type, label_type>{ cmd_parser.input_filename, { static_cast<real_type>(cmd_parser.lower), static_cast<real_type>(cmd_parser.upper) } } };
    }
}
 
template <typename cmd_parser_type>
[[nodiscard]] inline data_set_variants data_set_factory(const cmd_parser_type &cmd_parser) {
    if (cmd_parser.float_as_real_type && cmd_parser.strings_as_labels) {
        return data_set_factory_impl<float, std::string>(cmd_parser);
    } else if (cmd_parser.float_as_real_type && !cmd_parser.strings_as_labels) {
        return data_set_factory_impl<float>(cmd_parser);
    } else if (!cmd_parser.float_as_real_type && cmd_parser.strings_as_labels) {
        return data_set_factory_impl<double, std::string>(cmd_parser);
    } else {
        return data_set_factory_impl<double>(cmd_parser);
    }
}
}  // namespace plssvm::detail::cmd
 
#endif  // PLSSVM_DETAIL_CMD_DATA_SET_VARIANTS_HPP_