可以在Python中使用leastsq（）来解决多类问题吗？

我有两类值，希望使用最小二乘法进行分类。 理想情况下，我希望得到如下输出：

被该行分割的值属于两个给定的类

但是使用leastsq（）函数，我得到了如下结果：

似乎leastsq（）只接受值并进行计算。但是我想从不同的类中输入两组值，得到一个类似于第一个图的结果

有没有办法解决这种问题