跳至主要內容
逻辑回归的推导及实现

摘要

本文中,我们先推导了逻辑回归的公式,梯度下降过程;然后使用torch从零开始实现了一个LR model,接着使用torch自带函数来简化实现过程;然后在二分类数据集上演示了模型的训练效果。 最后我们介绍了两种在LR模型上使用多分类的思想,分别是ONE-vs-ALLONE-vs-ONE,并实现了这两种算法,在三分类数据集上测试了这两种方法,从结果来看效果都非常好。


yczha大约 13 分钟Machine Learning逻辑回归pytorch