pred_neg = (output<= 0.5).view(-1) pred_pos = (output> 0.5).view(-1) plt.scatter(x[pred_neg, 0], x[pred_neg, 1]) plt.scatter(x[pred_pos, 0], x[pred_pos, 1]) w=lr_model.liner.weight[0] b=lr_model.linear.bias[0] def draw_decision_boundary(w,b,x0): x1=(-b- w[0]* x0) /w[1] plt.plot(x0.detach().numpy(),x1.detach().numpy(),'r') draw_decision_boundary(w,b,torch.linspace(x. min(),x.max(),50))
时间: 2023-05-18 19:03:18 浏览: 135
Code.rar_PRED-163_matlab pred_社交网络_社交网络分析 链路预测_链路预测
5星 · 资源好评率100%
这是一个用于绘制逻辑回归决策边界的代码片段,其中 x 是输入数据,output 是模型的输出,lr_model 是逻辑回归模型。函数 draw_decision_boundary 用于绘制决策边界,其中 w 和 b 是模型的参数,x0 是决策边界上的点。
阅读全文