Expand description
rs-ml is a simple ML framework for the Rust language. it includes train test splitting, scalers, and a guassian naive bayes model. It also includes traits to add more transfomers and models to the framework.
Modules§
- classification
- Commonly used classification models and trait to implement more.
- metrics
- Commonly used metrics for classification and regression models.
- regression
- Regression
- transformer
- Abilities to transform and scale data.
Traits§
- Estimator
- Trait for fitting classification and regression models, and transformers.
Functions§
- train_
test_ split - Split data and features into training and testing set.
test_sizemust be between 0 and 1. Panics iftest_sizeis outside 0 and 1.