parser_predict.hpp

Go to the documentation of this file.

 
#ifndef PLSSVM_DETAIL_CMD_PARSER_PREDICT_HPP_
#define PLSSVM_DETAIL_CMD_PARSER_PREDICT_HPP_
#pragma once
 
#include "plssvm/backend_types.hpp"                      // plssvm::backend_type
#include "plssvm/backends/SYCL/implementation_type.hpp"  // plssvm::sycl::implementation_type
#include "plssvm/target_platforms.hpp"                   // plssvm::target_platform
 
#include <iosfwd>                                        // forward declare std::ostream
#include <string>                                        // std::string
 
namespace plssvm::detail::cmd {
 
class parser_predict {
  public:
    parser_predict(int argc, char **argv);
 
    backend_type backend{ backend_type::automatic };
    target_platform target{ target_platform::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 predict_filename{};
 
    std::string performance_tracking_filename{};
};
 
std::ostream &operator<<(std::ostream &out, const parser_predict &params);
 
}  // namespace plssvm::detail::cmd
 
#endif  // PLSSVM_DETAIL_CMD_PARSER_PREDICT_HPP_