SVM Classifier C++ 1.0.0
High-performance Support Vector Machine classifier with scikit-learn compatible API
|
One-vs-One (OvO) multiclass strategy. More...
#include <multiclass_strategy.hpp>
Public Member Functions | |
OneVsOneStrategy () | |
Constructor. | |
~OneVsOneStrategy () override | |
Destructor. | |
TrainingMetrics | fit (const torch::Tensor &X, const torch::Tensor &y, const KernelParameters ¶ms, DataConverter &converter) override |
Train the multiclass classifier. | |
std::vector< int > | predict (const torch::Tensor &X, DataConverter &converter) override |
Predict class labels. | |
std::vector< std::vector< double > > | predict_proba (const torch::Tensor &X, DataConverter &converter) override |
Predict class probabilities. | |
std::vector< std::vector< double > > | decision_function (const torch::Tensor &X, DataConverter &converter) override |
Get decision function values. | |
std::vector< int > | get_classes () const override |
Get unique class labels. | |
bool | supports_probability () const override |
Check if the model supports probability prediction. | |
int | get_n_classes () const override |
Get number of classes. | |
MulticlassStrategy | get_strategy_type () const override |
Get strategy type. | |
![]() | |
virtual | ~MulticlassStrategyBase ()=default |
Virtual destructor. | |
Additional Inherited Members | |
![]() | |
std::vector< int > | classes_ |
Unique class labels. | |
bool | is_trained_ = false |
Whether the model is trained. | |
One-vs-One (OvO) multiclass strategy.
Definition at line 171 of file multiclass_strategy.hpp.
|
overridevirtual |
Get decision function values.
X | Feature tensor of shape (n_samples, n_features) |
converter | Data converter instance |
Implements svm_classifier::MulticlassStrategyBase.
|
overridevirtual |
Train the multiclass classifier.
X | Feature tensor of shape (n_samples, n_features) |
y | Target tensor of shape (n_samples,) |
params | Kernel parameters |
converter | Data converter instance |
Implements svm_classifier::MulticlassStrategyBase.
|
inlineoverridevirtual |
Get unique class labels.
Implements svm_classifier::MulticlassStrategyBase.
Definition at line 197 of file multiclass_strategy.hpp.
|
inlineoverridevirtual |
Get number of classes.
Implements svm_classifier::MulticlassStrategyBase.
Definition at line 201 of file multiclass_strategy.hpp.
|
inlineoverridevirtual |
Get strategy type.
Implements svm_classifier::MulticlassStrategyBase.
Definition at line 203 of file multiclass_strategy.hpp.
|
overridevirtual |
Predict class labels.
X | Feature tensor of shape (n_samples, n_features) |
converter | Data converter instance |
Implements svm_classifier::MulticlassStrategyBase.
|
overridevirtual |
Predict class probabilities.
X | Feature tensor of shape (n_samples, n_features) |
converter | Data converter instance |
Implements svm_classifier::MulticlassStrategyBase.
|
overridevirtual |
Check if the model supports probability prediction.
Implements svm_classifier::MulticlassStrategyBase.