From e2c1db008aa05266f97e7f5f4e1ceb38003b6d13 Mon Sep 17 00:00:00 2001 From: LiuYuHui Date: Thu, 6 Aug 2020 10:48:37 +0800 Subject: [PATCH] Refactor all machines --- shogun/structure/DualLibQPBMSOSVM.cpp | 2 +- shogun/structure/DualLibQPBMSOSVM.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/shogun/structure/DualLibQPBMSOSVM.cpp b/shogun/structure/DualLibQPBMSOSVM.cpp index 19a1d4a..abf7cd6 100644 --- a/shogun/structure/DualLibQPBMSOSVM.cpp +++ b/shogun/structure/DualLibQPBMSOSVM.cpp @@ -86,7 +86,7 @@ void DualLibQPBMSOSVM::init() set_solver(BMRM); } -bool DualLibQPBMSOSVM::train_machine(std::shared_ptr data) +bool DualLibQPBMSOSVM::train_machine(const std::shared_ptr& data, const std::shared_ptr& labs) { require(m_model, "Model has not been set."); if (!m_w) diff --git a/shogun/structure/DualLibQPBMSOSVM.h b/shogun/structure/DualLibQPBMSOSVM.h index 6a65b99..f42f41d 100644 --- a/shogun/structure/DualLibQPBMSOSVM.h +++ b/shogun/structure/DualLibQPBMSOSVM.h @@ -233,7 +233,7 @@ class DualLibQPBMSOSVM : public LinearStructuredOutputMachine /** train dual SO-SVM * */ - bool train_machine(std::shared_ptr data=NULL) override; + bool train_machine(const std::shared_ptr& data, const std::shared_ptr& labs) override; private: /** init parameters