Summary
An easy to use and learn ML toolkit for Rust
Features
- Simple and intuitive API for common Machine Learning tasks.
- Implementations of popular algorithms like K-Nearest Neighbors and Decision Trees.
- Support for classification, regression, and clustering.
- Utility functions for data manipulation and metrics evaluation.
- Includes sample datasets like Iris, Housing, and Breast Cancer for quick experimentation.
Installation
Add Rusty Science to your Cargo.toml dependencies:
[]
= "0.1.1"
Usage
use KNNClassifier;
use load_iris;
Note: This crate is a work in progress and features are subject to change
Implementation table
Features:
| Feature | Implemented? |
|---|---|
| KNNClassifier | ✅ Implemented |
| KNNRegression | ✅ Implemented |
| KNNCluster | ✅ Implemented |
| Decision Tree Regression | ✅ Implemented |
| Decision tree Classifier | ✅ Implemented |
| Perceptron | ✅ Implemented |
| MLP Classifier | ❌ Not Implemented |
| MLP Regression | ❌ Not Implemented |
| Linear Regression | ✅ Implemented |
| Data Functions (train-test split) | ✅ Train test split |
| Dummy Datasets | ✅ Implemented |
| Graphing - Integrate the plotters crate? | ❌ Not Implemented |
| Binary SVC | ✅ Implemented |
| SVR | 🚧 Not Implemented |
| DBSCAN clustering | ✅ Implemented |
| Gaussian Mixture Model | ❌ Not Implemented |
| BIRCH algorithm | ❌ Not Implemented |
| Lasso Regression | ❌ Not Implemented |
| PCA | ❌ Not Implemented |
| Ridge Regression | ❌ Not Implemented |
| ElasticNet | ❌ Not Implemented |
| Lars | ❌ Not Implemented |
Metrics:
| Metric | Implemented |
|---|---|
| Accuracy | ✅ Implemented |
| r2 | ✅ Implemented |
| MAE | ✅ Implemented |
| MSE | ❌ Not Implemented |
| Precision | ❌ Not Implemented |
Datasets:
| Dataset | Implemented |
|---|---|
| Iris | ✅ Implemented |
| Housing | ✅ Implemented |
| Brest Cancer | ✅ Implemented |
Contact
If you want to contact us email us at cooper.brown197@gmail.com or jack.welsh@drake.edu