csv_dataset.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <vector>
 
#include "pulsatrix/csv_reader.hpp"
#include "pulsatrix/dataset.hpp"
#include "pulsatrix/device_backend.hpp"
 
namespace pulsatrix {
 
class CsvDataset : public Dataset {
public:
    CsvDataset(const std::string& path, std::vector<std::string> feature_columns, std::string label_column,
               DeviceBackend* backend);
 
    [[nodiscard]] int64_t size() const override;
 
    [[nodiscard]] Sample get(int64_t index) const override;
 
private:
    CsvTable table_;
    std::vector<size_t> feature_col_indices_;
    size_t label_col_index_;
    DeviceBackend* backend_;
};
 
}  // namespace pulsatrix