sparse_svm_function.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_SPARSE_SVM_SPARSE_SVM_FUNCTION_HPP
#define MLPACK_METHODS_SPARSE_SVM_SPARSE_SVM_FUNCTION_HPP

#include <mlpack/prereqs.hpp>

class SparseSVMFunction
{
 public:
  SparseSVMFunction() {}

  SparseSVMFunction(const arma::sp_mat& dataset, const arma::vec& labels);

  void Shuffle();

  double Evaluate(const arma::mat& parameters,
                  const size_t startId,
                  const size_t batchSize = 1);

  template <typename GradType>
  void Gradient(const arma::mat& parameters, size_t id, GradType& gradient);

  const arma::mat& InitialPoint() const { return initialPoint; }
  arma::mat& InitialPoint() { return initialPoint; }

  const arma::sp_mat& Dataset() const { return dataset; }
  arma::sp_mat& Dataset() { return dataset; }

  const arma::vec& Labels() const { return labels; }
  arma::vec& Labels() { return labels; }

  size_t NumFunctions();

 private:
  arma::mat initialPoint;

  arma::sp_mat dataset;

  arma::vec labels;
};

// Include implementation
#include "sparse_svm_function_impl.hpp"

#endif // MLPACK_METHODS_SPARSE_SVM_SPARSE_SVM_FUNCTION_HPP