svm_parameter.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <cuml/common/logger.hpp>
 
namespace ML {
namespace SVM {
 
enum SvmType { C_SVC, NU_SVC, EPSILON_SVR, NU_SVR };
 
struct SvmParameter {
  double C;           
  double cache_size;  
  int max_outer_iter;
  int max_iter;
  int nochange_steps;                   //<! Number of steps to continue with non-changing diff
  double tol;                           
  rapids_logger::level_enum verbosity;  
  double epsilon;                       
  SvmType svmType;
};
 
};  // namespace SVM
};  // namespace ML