sampler.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
#include <optional>
#include <random>
#include <vector>
 
namespace pulsatrix {
 
class Sampler {
public:
    virtual ~Sampler() = default;
 
    virtual void reset(int64_t dataset_size) = 0;
 
    [[nodiscard]] virtual std::optional<int64_t> next() = 0;
};
 
class SequentialSampler : public Sampler {
public:
    void reset(int64_t dataset_size) override;
    [[nodiscard]] std::optional<int64_t> next() override;
 
private:
    int64_t size_ = 0;
    int64_t position_ = 0;
};
 
class ShuffleSampler : public Sampler {
public:
    explicit ShuffleSampler(unsigned seed);
    void reset(int64_t dataset_size) override;
    [[nodiscard]] std::optional<int64_t> next() override;
 
private:
    unsigned seed_;
    std::mt19937 rng_;
    std::vector<int64_t> indices_;
    size_t position_ = 0;
};
 
}  // namespace pulsatrix