rcf3 0.5.0

A Rust implementation of the Random Cut Forest algorithm for anomaly detection.
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
49
50
51

#![cfg(feature = "python")]

use pyo3::exceptions::{PyRuntimeError, PyValueError};
use pyo3::prelude::*;
use std::collections::BTreeMap;

use crate::error::RcfError;

/// Convert an [`RcfError`] to a Python exception.
pub(crate) fn to_py_err(e: RcfError) -> PyErr {
    match e {
        RcfError::InvalidArgument(msg) => PyValueError::new_err(msg),
        RcfError::DimensionMismatch { expected, got } => PyValueError::new_err(format!(
            "dimension mismatch: expected {expected}, got {got}"
        )),
        other => PyRuntimeError::new_err(other.to_string()),
    }
}

#[derive(FromPyObject)]
pub(crate) enum StrOrBytes {
    Str(String),
    Bytes(Vec<u8>),
}

impl AsRef<[u8]> for StrOrBytes {
    fn as_ref(&self) -> &[u8] {
        match self {
            StrOrBytes::Str(s) => s.as_bytes(),
            StrOrBytes::Bytes(b) => b,
        }
    }
}

impl From<String> for StrOrBytes {
    fn from(value: String) -> Self {
        Self::Str(value)
    }
}

impl From<Vec<u8>> for StrOrBytes {
    fn from(value: Vec<u8>) -> Self {
        Self::Bytes(value)
    }
}

#[derive(FromPyObject)]
pub(crate) enum KeyValueLike {
    Pairs(Vec<(String, f64)>),
    Dict(BTreeMap<String, f64>),
}