svr.hpp

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/matrix/kernel_params.hpp>
 
#include <cublas_v2.h>
 
namespace ML {
namespace SVM {
 
template <typename math_t>
struct SvmModel;
struct SvmParameter;
 
// Forward declarations of the stateless API
template <typename math_t>
int svrFit(const raft::handle_t& handle,
           math_t* X,
           int n_rows,
           int n_cols,
           math_t* y,
           const SvmParameter& param,
           ML::matrix::KernelParams& kernel_params,
           SvmModel<math_t>& model,
           const math_t* sample_weight = nullptr);
 
template <typename math_t>
int svrFitSparse(const raft::handle_t& handle,
                 int* indptr,
                 int* indices,
                 math_t* data,
                 int n_rows,
                 int n_cols,
                 int nnz,
                 math_t* y,
                 const SvmParameter& param,
                 ML::matrix::KernelParams& kernel_params,
                 SvmModel<math_t>& model,
                 const math_t* sample_weight = nullptr);
 
// For prediction we use svcPredict
 
};  // end namespace SVM
};  // end namespace ML