ip_metric.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_FASTMKS_IP_METRIC_HPP
#define MLPACK_METHODS_FASTMKS_IP_METRIC_HPP

namespace mlpack {
namespace metric {

template<typename KernelType>
class IPMetric
{
 public:
  IPMetric();

  IPMetric(KernelType& kernel);

  ~IPMetric();

  template<typename VecTypeA, typename VecTypeB>
  typename VecTypeA::elem_type Evaluate(const VecTypeA& a, const VecTypeB& b);

  const KernelType& Kernel() const { return *kernel; }
  KernelType& Kernel() { return *kernel; }

  template<typename Archive>
  void Serialize(Archive& ar, const unsigned int version);

 private:
  KernelType* kernel;
  bool kernelOwner;
};

} // namespace metric
} // namespace mlpack

// Include implementation.
#include "ip_metric_impl.hpp"

#endif