dataset.hpp

Go to the documentation of this file.

#pragma once
 
#include <pineforge/pineforge.h>
 
#include <cstddef>
#include <filesystem>
#include <vector>
 
namespace pineforge {
namespace hpo {
 
class Dataset final {
public:
    explicit Dataset(std::vector<pf_bar_t> bars);
 
    static Dataset load_csv(const std::filesystem::path& path);
 
    const pf_bar_t* data() const noexcept { return bars_.data(); }
    std::size_t size() const noexcept { return bars_.size(); }
    bool empty() const noexcept { return bars_.empty(); }
    const std::vector<pf_bar_t>& bars() const noexcept { return bars_; }
 
private:
    std::vector<pf_bar_t> bars_;
};
 
}  // namespace hpo
}  // namespace pineforge