Skip to main content

_ferrolearn_rs/
lib.rs

1mod classifiers;
2mod clusterers;
3mod conversions;
4mod regressors;
5mod transformers;
6
7use pyo3::prelude::*;
8
9#[pymodule]
10fn _ferrolearn_rs(m: &Bound<'_, PyModule>) -> PyResult<()> {
11    // Regressors
12    m.add_class::<regressors::RsLinearRegression>()?;
13    m.add_class::<regressors::RsRidge>()?;
14    m.add_class::<regressors::RsLasso>()?;
15    m.add_class::<regressors::RsElasticNet>()?;
16
17    // Classifiers
18    m.add_class::<classifiers::RsLogisticRegression>()?;
19    m.add_class::<classifiers::RsDecisionTreeClassifier>()?;
20    m.add_class::<classifiers::RsRandomForestClassifier>()?;
21    m.add_class::<classifiers::RsKNeighborsClassifier>()?;
22    m.add_class::<classifiers::RsGaussianNB>()?;
23
24    // Transformers
25    m.add_class::<transformers::RsStandardScaler>()?;
26    m.add_class::<transformers::RsPCA>()?;
27
28    // Clusterers
29    m.add_class::<clusterers::RsKMeans>()?;
30
31    Ok(())
32}