plot_data.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cstdint>
#include <string>
#include <vector>
 
#include "pulsatrix/attribution.hpp"
#include "pulsatrix/circuit_graph.hpp"
#include "pulsatrix/dataset.hpp"
 
namespace pulsatrix {
 
struct BarSeries {
    std::vector<std::string> labels;
    std::vector<float> values;
};
 
[[nodiscard]] BarSeries ToFeatureImportanceBars(const Attribution& attr, int top_k);
 
struct WaterfallStep {
    std::string label;
    float delta;
    float cumulative;
};
 
[[nodiscard]] std::vector<WaterfallStep> ToWaterfallSteps(const Attribution& attr, float baseline_value);
 
struct WaterfallBar {
    float bottom;
    float top;
    bool increase;
};
 
[[nodiscard]] std::vector<WaterfallBar> ToWaterfallBars(const std::vector<WaterfallStep>& steps, float baseline_value);
 
struct HeatmapGrid {
    std::vector<float> values;
    int64_t rows;
    int64_t cols;
};
 
[[nodiscard]] HeatmapGrid ToSaliencyHeatmap(const Attribution& attr);
 
struct HeatmapColorScale {
    float scale_min;
    float scale_max;
    bool is_signed;
};
 
[[nodiscard]] HeatmapColorScale ComputeHeatmapColorScale(const HeatmapGrid& grid);
 
struct BeeswarmPoint {
    float x;
    float y;
};
 
[[nodiscard]] std::vector<BeeswarmPoint> ToBeeswarmPoints(const std::vector<Attribution>& runs, int64_t feature_index);
 
[[nodiscard]] std::string CircuitNodeDisplayLabel(const CircuitNode& node);
 
struct HistogramBins {
    std::vector<float> bin_edges;
    std::vector<int64_t> counts;
};
 
[[nodiscard]] HistogramBins ToFieldHistogramBins(const Dataset& dataset, int64_t field_index, int num_bins);
 
struct RgbImageBuffer {
    std::vector<unsigned char> pixels;
    int64_t height;
    int64_t width;
};
 
[[nodiscard]] RgbImageBuffer ToRgbImageBuffer(const Tensor& image_chw);
 
}  // namespace pulsatrix