#linear #train #model #learning #machine-learning #weight #tangram

bin+lib tangram_linear

Tangram 使得程序员能够轻松地训练、部署和监控机器学习模型

4 个版本 (破坏性更新)

0.7.0 2021 年 8 月 17 日
0.6.0 2021 年 7 月 19 日
0.5.0 2021 年 7 月 2 日
0.4.0 2021 年 6 月 25 日

机器学习 分类中排名 #505

每月下载量 28
3 个crate中使用(直接使用2个)

MIT 许可证 property="license">MIT

340KB
7.5K SLoC

Rust 6.5K SLoC // 0.0% comments Python 821 SLoC // 0.0% comments

拼图线性

此crate实现了用于回归和分类的线性机器学习模型。有三种模型类型,RegressorBinaryClassifierMulticlassClassifier。`BinaryClassifier` 使用sigmoid激活函数,`MulticlassClassifier` 训练`n_classes`个线性模型,其输出通过softmax函数组合。

为了在多核处理器上加快训练速度,我们允许从多个线程同时读写模型参数。这意味着每个线程都将读取其他线程部分更新的权重,它写入的权重可能被其他线程覆盖。这使得训练非确定性,但在实践中,我们观察到结果变化很小,因为有反馈控制:在每个epoch后监控损失的变化,当损失稳定时停止训练。

依赖项

~12MB
~209K SLoC