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个)
340KB
7.5K SLoC
拼图线性
此crate实现了用于回归和分类的线性机器学习模型。有三种模型类型,Regressor
、BinaryClassifier
和 MulticlassClassifier
。`BinaryClassifier` 使用sigmoid激活函数,`MulticlassClassifier` 训练`n_classes`个线性模型,其输出通过softmax函数组合。
为了在多核处理器上加快训练速度,我们允许从多个线程同时读写模型参数。这意味着每个线程都将读取其他线程部分更新的权重,它写入的权重可能被其他线程覆盖。这使得训练非确定性,但在实践中,我们观察到结果变化很小,因为有反馈控制:在每个epoch后监控损失的变化,当损失稳定时停止训练。
依赖项
~12MB
~209K SLoC