zhangxiann / PyTorch_Practice

这是我学习 PyTorch 的笔记对应的代码,点击查看 PyTorch 笔记在线电子书
http://pytorch.zhangxiann.com/
GNU General Public License v3.0
1.24k stars 273 forks source link

请问lesson1 逻辑回归里面这行代码的数学原理是什么? #4

Open wongsihan opened 3 years ago

wongsihan commented 3 years ago

plot_y = (-w0 * plot_x - plot_b) / w1

这里plot画出的是两种类别的分割线,没明白y为什么要这么取值,nn.linear的两个权重w0 w1分别有什么含义呢?

zhangxiann commented 3 years ago

你可以反推一下: plot_y = (-w0 * plot_x - plot_b) / w1 可以化为 w1 * plot_y + w0 * plot_x + plot_b = 0

这就是逻辑回归的公式中的 f(x) image 附上逻辑回归公式: image