#include <svm_model.h>
Collaboration diagram for ML::SVM::SvmModel< math_t >:
Public Attributes | |
| int | n_support |
| Number of support vectors. More... | |
| int | n_cols |
| Number of features. More... | |
| math_t | b |
| Constant used in the decision function. More... | |
| math_t * | dual_coefs |
| SupportStorage< math_t > | support_matrix |
| Support vector storage - can contain either CSR or dense. More... | |
| int * | support_idx |
| Indices (from the training set) of the support vectors, size [n_support]. More... | |
| int | n_classes |
| math_t * | unique_labels |
| Device pointer for the unique classes. Size [n_classes]. More... | |
Detailed Description
template<typename math_t>
struct ML::SVM::SvmModel< math_t >
Parameters that describe a trained SVM model. All pointers are device pointers.
Member Data Documentation
◆ b
template<typename math_t >
| math_t ML::SVM::SvmModel< math_t >::b |
Constant used in the decision function.
◆ dual_coefs
template<typename math_t >
| math_t* ML::SVM::SvmModel< math_t >::dual_coefs |
Non-zero dual coefficients ( dual_coef[i] = \( y_i \alpha_i \)). Size [n_support].
◆ n_classes
template<typename math_t >
| int ML::SVM::SvmModel< math_t >::n_classes |
Number of classes found in the input labels
◆ n_cols
template<typename math_t >
| int ML::SVM::SvmModel< math_t >::n_cols |
Number of features.
◆ n_support
template<typename math_t >
| int ML::SVM::SvmModel< math_t >::n_support |
Number of support vectors.
◆ support_idx
template<typename math_t >
| int* ML::SVM::SvmModel< math_t >::support_idx |
Indices (from the training set) of the support vectors, size [n_support].
◆ support_matrix
template<typename math_t >
| SupportStorage<math_t> ML::SVM::SvmModel< math_t >::support_matrix |
Support vector storage - can contain either CSR or dense.
◆ unique_labels
template<typename math_t >
| math_t* ML::SVM::SvmModel< math_t >::unique_labels |
Device pointer for the unique classes. Size [n_classes].
The documentation for this struct was generated from the following file:
- include/cuml/svm/svm_model.h
