parser_train.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_DETAIL_CMD_PARSER_TRAIN_HPP_
#define PLSSVM_DETAIL_CMD_PARSER_TRAIN_HPP_
#pragma once
 
#include "plssvm/backend_types.hpp"                         // plssvm::backend_type
#include "plssvm/backends/SYCL/implementation_type.hpp"     // plssvm::sycl::implementation_type
#include "plssvm/backends/SYCL/kernel_invocation_type.hpp"  // plssvm::sycl::kernel_invocation_type
#include "plssvm/default_value.hpp"                         // plssvm::default_value
#include "plssvm/parameter.hpp"                             // plssvm::parameter
#include "plssvm/target_platforms.hpp"                      // plssvm::target_platform
 
#include <cstddef>                                          // std::size_t
#include <iosfwd>                                           // forward declare std::ostream
#include <string>                                           // std::string
 
namespace plssvm::detail::cmd {
 
class parser_train {
  public:
    parser_train(int argc, char **argv);
 
    plssvm::parameter csvm_params{};
 
    default_value<double> epsilon{ default_init<double>{ 0.001 } };
    default_value<std::size_t> max_iter{ default_init<std::size_t>{ 0 } };
 
    backend_type backend{ backend_type::automatic };
    target_platform target{ target_platform::automatic };
 
    sycl::kernel_invocation_type sycl_kernel_invocation_type{ sycl::kernel_invocation_type::automatic };
    sycl::implementation_type sycl_implementation_type{ sycl::implementation_type::automatic };
 
    bool strings_as_labels{ false };
    bool float_as_real_type{ false };
 
    std::string input_filename{};
    std::string model_filename{};
 
    std::string performance_tracking_filename{};
};
 
std::ostream &operator<<(std::ostream &out, const parser_train &params);
 
}  // namespace plssvm::detail::cmd
 
#endif  // PLSSVM_DETAIL_CMD_PARSER_TRAIN_HPP_