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