use serde::{Deserialize, Serialize};
use std::path::Path;
use vega_lite_4::*;
#[derive(Serialize, Deserialize)]
pub struct Item {
pub x: f64,
pub y: f64,
pub cluster: usize,
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let mut rdr = csv::Reader::from_path(Path::new("examples/res/data/clustered_data.csv"))?;
let values = rdr
.deserialize()
.collect::<Result<Vec<Item>, csv::Error>>()?;
let chart = VegaliteBuilder::default()
.title("Clusters")
.description("Dots colored by their cluster.")
.data(&values)
.mark(Mark::Point)
.encoding(
EdEncodingBuilder::default()
.x(XClassBuilder::default()
.field("x")
.position_def_type(Type::Quantitative)
.build()?)
.y(YClassBuilder::default()
.field("y")
.position_def_type(Type::Quantitative)
.build()?)
.color(ColorClassBuilder::default().field("cluster").build()?)
.build()?,
)
.build()?;
chart.show()?;
eprint!("{}", chart.to_string()?);
Ok(())
}