From 14856930a79b4246bff951330e56200ba043d34d Mon Sep 17 00:00:00 2001 From: manzerbredes Date: Sat, 20 Feb 2021 20:36:33 +0100 Subject: Add more details to the logistic regression --- logistic_regression/binary.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'logistic_regression/binary.py') diff --git a/logistic_regression/binary.py b/logistic_regression/binary.py index a8a11e2..1c3f608 100755 --- a/logistic_regression/binary.py +++ b/logistic_regression/binary.py @@ -97,6 +97,10 @@ scatter=plt.scatter(x_1,x_2,c=np.round(h(x_1,x_2)),marker="o") handles, labels = scatter.legend_elements(prop="colors", alpha=0.6) legend = ax.legend(handles, ["Class A","Class B"], loc="upper right", title="Legend") +x=np.arange(0,10,0.2) +plt.plot([1,2],[2,2]) + # Save plt.tight_layout() -plt.savefig("binary.png",dpi=300) +#plt.savefig("binary.png",dpi=300) +plt.show() -- cgit v1.2.3