SomeB1oody/RustyML
A high-performance machine learning library in pure Rust, offering statistical utilities, ML algorithms and neural networks.
解決的問題
RustyML 是一個純 Rust 實作的函式庫,提供完整的機器學習與深度學習工作流程。它消除了對 C 或 C++ 依賴(FFI)的需求,為在 Rust 中實作 ML 模型提供了一種記憶體安全、可移植且高效率的替代方案。
如何運作
該函式庫採用模組化功能旗標設計,允許使用者僅包含所需的元件。它利用 ndarray crate 進行張量運算,並使用 Rayon 實現重型內核的多執行緒平行化。支援兩種類型的神經網路架構:用於線性堆疊層的 Sequential 模型,以及用於複雜架構(如殘差連接或多輸入/輸出網路)的 Graph 模型。
適用對象
希望使用純 Rust 生態系統建構與部署機器學習模型的開發者,避免連結外部 C/C++ 庫的複雜性,同時維持記憶體安全與平行處理能力。
主要特色
- 純 Rust 實作:無 FFI 或外部 C/C++ 依賴,提升可移植性與記憶體安全性。
- 廣泛的演算法涵蓋:包含傳統的監督/非監督學習、異常偵測,以及完整的神經網路框架。
- 彈性的神經網路架構:支援順序模型與圖形模型(允許實作殘差模組與共享塔結構)。
- 決定性結果:提供全域與元件級種子管理,確保實驗可重現。
- 模型持久化:使用
Serde與postcard將訓練完成的模型儲存與載入為緊湊的二進位檔案。 - 與 Scikit-learn 相容的評估指標:包含回歸、分類與群聚任務的完整評估指標套件。
相關
- 專案
- 專案
- 專案
- 專案
- 專案