apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding the missing template parameter

Browse Source 
Signed-off-by: Omar Shrit <omar@avontech.fr>
This commit is contained in:
Omar Shrit
2024-04-16 18:51:35 +02:00
parent abda56982e
commit ef7779f27f
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/bias_svd/bias_svd_impl.hpp
+9 -9
View File
@@ -17,9 +17,9 @@
namespace mlpack {
template<typename OptimizerType, typename MatType, typename VecType>
BiasSVD<OptimizerType>::BiasSVD(const size_t iterations,
const double alpha,
const double lambda) :
BiasSVD<OptimizerType, MatType, VecType>::BiasSVD(const size_t iterations,
const double alpha,
const double lambda) :
iterations(iterations),
alpha(alpha),
lambda(lambda)
@@ -28,12 +28,12 @@ BiasSVD<OptimizerType>::BiasSVD(const size_t iterations,
}
template<typename OptimizerType, typename MatType, typename VecType>
void BiasSVD<OptimizerType>::Apply(const MatType& data,
const size_t rank,
MatType& u,
MatType& v,
VecType& p,
VecType& q)
void BiasSVD<OptimizerType, MatType, VecType>::Apply(const MatType& data,
const size_t rank,
MatType& u,
MatType& v,
VecType& p,
VecType& q)
{
// batchSize is 1 in our implementation of Bias SVD.
// batchSize other than 1 has not been supported yet.