0

我想为 C++ 上具有 5 个参数的函数编写一个简单的梯度下降。现在我偶然发现了一个实现模型问题:我是否应该牺牲速度来将我的参数折叠到向量/数组中。这就是我的意思。我可以像这样实现函数值和梯度计算:

double function(double arg1, double arg2, ..., double arg5);
double functionGradient1(double arg1, double arg2, ..., double arg5);
double functionGradient2(double arg1, double arg2, ..., double arg5);
...
double functionGradient5(double arg1, double arg2, ..., double arg5);

或者:

double function(double * args);
double functionGradientAt(double * args, int i);

最后一个更容易实现,但是如果我要不断分配/释放内存,恐怕我会失去很多速度。

我的问题是:

  1. 你会选择哪一个?
  2. 如果你曾经实现过梯度下降,你是如何处理这种情况的?
4

2 回答 2

1

你会选择哪一个?

我会选择:

double function(const vector<double> &args);

或者,如果它必须来自您的示例,那么:

double function(double * args);

原因是我不相信展开这样的论点会增加任何明显的加速,但它确实使代码看起来很难看。

如果你曾经实现过梯度下降,你是如何处理这种情况的?

我用向量处理它。

梯度下降的速度可能更多地取决于您执行的迭代次数,而不是您传递这些参数的方式。如果您想尽可能快,我建议您不要自己实现它,而是找到一个实现它的库。这个问题有很多很好的建议。

于 2015-03-09T10:38:23.923 回答
0

在这两种情况下: double functionGradient5(double arg1, double arg2, ..., double arg5); 以及双功能(双* args);您可能需要分配相同数量的内存。第一个解决方案将在堆中分配内存,第二个你必须自己处理它,这样对你来说不太方便,但另一方面你可以遍历参数,所以在某些情况下它可以会更好。

于 2015-03-09T10:38:25.147 回答