sklears-semi-supervised 0.1.0

Semi-supervised learning algorithms
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
41
42
43
44
45
46
47
48

//! Batch active learning methods for semi-supervised learning
//!
//! This module implements various batch active learning strategies that select
//! multiple samples simultaneously for labeling, considering both uncertainty
//! and diversity to create informative batches.

mod batch_mode;
mod core_set;
mod diverse_minibatch;
mod diversity_based;
mod gradient_embedding;

pub use batch_mode::*;
pub use core_set::*;
pub use diverse_minibatch::*;
pub use diversity_based::*;
pub use gradient_embedding::*;

use scirs2_core::ndarray_ext::{Array1, Array2, ArrayView1, ArrayView2};
use scirs2_core::random::Random;
// use scirs2_core::random::rand::seq::SliceRandom as _; // Bring trait methods into scope
use sklears_core::error::{Result, SklearsError};
use std::collections::HashMap;
use thiserror::Error;

#[derive(Error, Debug)]
pub enum BatchActiveLearningError {
    #[error("Invalid batch size: {0}")]
    InvalidBatchSize(usize),
    #[error("Invalid diversity weight: {0}")]
    InvalidDiversityWeight(f64),
    #[error("Invalid cluster count: {0}")]
    InvalidClusterCount(usize),
    #[error("Insufficient unlabeled samples")]
    InsufficientUnlabeledSamples,
    #[error("Invalid distance metric: {0}")]
    InvalidDistanceMetric(String),
    #[error("Matrix operation failed: {0}")]
    MatrixOperationFailed(String),
    #[error("Core-set computation failed: {0}")]
    CoreSetComputationFailed(String),
}

impl From<BatchActiveLearningError> for SklearsError {
    fn from(err: BatchActiveLearningError) -> Self {
        SklearsError::FitError(err.to_string())
    }
}