首页 > 解决方案 > 如何在 C++ 中使用离散分布作为类变量

问题描述

所以我有一个由其他人开发的巨大程序(几个 .cpp 文件和 .h),我试图在其中一个文件中添加一些功能。我是 C++ 新手,所以对我来说很难。

我想在初始化“优化器”时使用初始化幂律分布,然后在调用“迭代”时从中采样。我已经设法让它工作,但只在“初始化”内部,我希望它作为一个类变量在整个文件中工作。我想要的最小示例如下,考虑到优化器是在“Optimizer.h”中定义的,并不是真正相关的,随机引擎是在“Util.h”中定义的。

头文件如下所示:

#ifndef FASTONEPLUSONE_H_
#define FASTONEPLUSONE_H_

#include "Optimizer.h"
#include "Util.h"
#include <map>

// Inherits and implements the Optimizer interface
class FastOnePlusOne : public Optimizer {
 public:
  FastOnePlusOne(Random& _rand, shared_ptr<Evaluator> _evaluator,
               Configuration& _config);
  virtual bool iterate() override;
  create_optimizer(FastOnePlusOne);

 private:
  // Power law distribution
  std::discrete_distribution<int>power_dist; // THIS IS MADE BY ME
};

#endif /* FASTONEPLUSONE_H_ */

和 cpp 文件:

FastOnePlusOne::FastOnePlusOne(Random& _rand, shared_ptr<Evaluator> _evaluator,
                           Configuration& _config)
    : Optimizer(_rand, _evaluator, _config) {
  // Power-law distribution
  int length = 10;
  float B = 1.5;
  vector<double> pmf(length);
  std::iota(pmf.begin(), pmf.end(), 1);
  for(size_t i = 0; i < length; i++) {
    pmf[i] = 1.0 / pow(pmf[i], B);
  }
  pmf.insert(pmf.begin(), 0);
  std::discrete_distribution<int> power_dist(pmf.begin(),pmf.end());
  
  std::map<int, int> m;
  for(int n=0; n<10000; ++n) {
    ++m[power_dist(rand)];
  }
  for(auto p : m) {
    std::cout << p.first << " generated " << p.second << " times\n";
  }
}

// Performs an iteration
bool FastOnePlusOne::iterate() {
  // Distributions needed at the current alpha
  int alpha = power_dist(rand);
  std::cout << alpha << " ";
  }
}

此代码现在在FastOnePlusOne初始化时在变量 power_dist 中创建幂律分布,它正确打印 10000 个样本的结果,但alpha始终采样为 0。我该如何解决这个问题?

标签: c++headerdistribution

解决方案


替换这一行:

std::discrete_distribution<int> power_dist(pmf.begin(),pmf.end());

有了这个:

power_dist = std::discrete_distribution<int>(pmf.begin(),pmf.end());

解释:

在错误的版本中,您正在创建一个名为power_distwho shadows the class variable的新局部变量this->power_dist,并使用它。Originalthis->power_dist从未在内部使用或正确初始化FastOnePlusOne::FastOnePlusOne()。在函数FastOnePlusOne::iterate()中,您对未初始化的this->power_dist.

在第二个版本中,您将为类变量分配一个正确初始化的对象,该变量power_dist稍后将用于同一函数和iterate()函数中。


推荐阅读