csv_loading/
csv_loading.rs

1use scirs2_datasets::loaders::load_csv;
2
3fn main() -> Result<(), Box<dyn std::error::Error>> {
4    // Load a CSV file with headers and target column
5    let dataset = load_csv(
6        "scirs2-datasets/data/example.csv",
7        true,    // has header
8        Some(3), // target column index (0-based)
9    )?;
10
11    println!("CSV dataset loaded successfully:");
12    println!("  Samples: {}", dataset.n_samples());
13    println!("  Features: {}", dataset.n_features());
14    println!("  Feature names: {:?}", dataset.feature_names);
15
16    // Access data and target
17    println!("\nFirst 3 samples:");
18    for i in 0..3 {
19        let features = dataset.data.row(i);
20        let target = dataset.target.as_ref().map(|t| t[i]);
21        println!(
22            "  Sample {}: Features = {:?}, Target = {:?}",
23            i, features, target
24        );
25    }
26
27    Ok(())
28}