tenflowers-dataset 0.1.1

Data pipeline and dataset utilities for TenfloweRS
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

//! Extension traits for dataset statistics
//!
//! This module provides convenient trait extensions that allow datasets
//! to compute statistics directly via method calls.

use crate::Dataset;
use tenflowers_core::Result;

use super::computation::DatasetStatisticsComputer;
use super::core::{DatasetStats, StatisticsConfig};

/// Extension trait for Dataset to add statistics computation
pub trait DatasetStatisticsExt<T> {
    /// Compute statistics with default configuration
    fn compute_statistics(&self) -> Result<DatasetStats<T>>;

    /// Compute statistics with custom configuration
    fn compute_statistics_with_config(&self, config: StatisticsConfig) -> Result<DatasetStats<T>>;
}

impl<T, D> DatasetStatisticsExt<T> for D
where
    T: Clone
        + Default
        + scirs2_core::numeric::Zero
        + scirs2_core::numeric::Float
        + std::fmt::Debug
        + Send
        + Sync
        + 'static,
    D: Dataset<T>,
{
    fn compute_statistics(&self) -> Result<DatasetStats<T>> {
        DatasetStatisticsComputer::compute(self, StatisticsConfig::default())
    }

    fn compute_statistics_with_config(&self, config: StatisticsConfig) -> Result<DatasetStats<T>> {
        DatasetStatisticsComputer::compute(self, config)
    }
}