NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added overloads for dot() so you can dot() a std::vector with a std::map.

Browse Source
This commit is contained in:
Davis King 2012-05-01 18:10:33 -04:00
parent 440c6571aa
commit 2f2aecc9fa
2 changed files with 82 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
dlib/svm/sparse_vector.h
View File

@ -7,6 +7,8 @@
#include <cmath>
#include <limits>
#include "../algs.h"
#include <vector>
#include <map>
namespace dlib
@ -216,37 +218,67 @@ namespace dlib
// ------------------------------------------------------------------------------------
namespace impl
{
template <typename T, typename U>
typename T::value_type::second_type dot (
const T& a,
const U& b
)
{
typedef typename T::value_type::second_type scalar_type;
typename T::const_iterator ai = a.begin();
typename U::const_iterator bi = b.begin();
scalar_type sum = 0;
while (ai != a.end() && bi != b.end())
{
if (ai->first == bi->first)
{
sum += ai->second * bi->second;
++ai;
++bi;
}
else if (ai->first < bi->first)
{
++ai;
}
else
{
++bi;
}
}
return sum;
}
}
template <typename T>
typename T::value_type::second_type dot (
inline typename T::value_type::second_type dot (
const T& a,
const T& b
)
{
typedef typename T::value_type::second_type scalar_type;
return dlib::sparse_vector::impl::dot(a,b);
}
typename T::const_iterator ai = a.begin();
typename T::const_iterator bi = b.begin();
template <typename T1, typename T2, typename T3, typename T4, typename T5, typename T6>
inline T4 dot (
const std::vector<T1,T2>& a,
const std::map<T3,T4,T5,T6>& b
)
{
return dlib::sparse_vector::impl::dot(a,b);
}
scalar_type sum = 0;
while (ai != a.end() && bi != b.end())
{
if (ai->first == bi->first)
{
sum += ai->second * bi->second;
++ai;
++bi;
}
else if (ai->first < bi->first)
{
++ai;
}
else
{
++bi;
}
}
return sum;
template <typename T1, typename T2, typename T3, typename T4, typename T5, typename T6>
inline T4 dot (
const std::map<T3,T4,T5,T6>& a,
const std::vector<T1,T2>& b
)
{
return dlib::sparse_vector::impl::dot(a,b);
}
// ------------------------------------------------------------------------------------

26
dlib/svm/sparse_vector_abstract.h
View File

@ -6,6 +6,8 @@
#include <cmath>
#include "../algs.h"
#include "../serialize.h"
#include <map>
#include <vector>
namespace dlib
{
@ -164,6 +166,30 @@ namespace dlib
- returns the dot product between the vectors a and b
!*/
template <typename T1, typename T2, typename T3, typename T4, typename T5, typename T6>
T4 dot (
const std::vector<T1,T2>& a,
const std::map<T3,T4,T5,T6>& b
);
/*!
requires
- a and b are sparse vectors
ensures
- returns the dot product between the vectors a and b
!*/
template <typename T1, typename T2, typename T3, typename T4, typename T5, typename T6>
T4 dot (
const std::map<T3,T4,T5,T6>& a,
const std::vector<T1,T2>& b
);
/*!
requires
- a and b are sparse vectors
ensures
- returns the dot product between the vectors a and b
!*/
// ----------------------------------------------------------------------------------------
template <typename T, typename EXP>