pytorch二分类

用神经网络解决一个简单的二类分类问题 仍然是看莫烦pytorch的教程。先上代码 fromtorch.autogradimportVariableimportmatplotlib.pyplotaspltimporttorch.nn.functionalasF#激励函数都在这里#假数据n_data=t

数据格式如下，第一列为标签，分别有0和1两个值；第2到4行分别为输入x的特征，也就是x有三个特征。 代码如下： importtorch# torch 是一种科学计算框架importtorch.nnasnn# torch.nn 神经网络的接口importnumpyasnp# numpy 科学计算的软件

同样是吴恩达深度学习课程的作业，我把它改成pytorch版本了。有了上次的经验，这次就没花多少时间了。 1 packages importnumpyasnpimporttorchfromtorchimportnnimportmatplotlib.pyplotaspltimporttorch.util